Updated generate to handle model input

Author: rupeshs

src/app.py

@@ -32,11 +32,15 @@ def _get_model(model_id: str) -> str:
     )
     args = parser.parse_args()
     compute = Computing()
-    generate = Generate(compute)
     model_id = _get_model(args.model)
 
     print(f"Model : {model_id}")
 
+    generate = Generate(
+        compute,
+        model_id,
+    )
+
     dm_web_ui = diffusionmagic_web_ui(
         generate,
         model_id,

src/backend/generate.py

@@ -22,7 +22,11 @@
 
 
 class Generate:
-    def __init__(self, compute: Computing):
+    def __init__(
+        self,
+        compute: Computing,
+        model_id: str,
+    ):
         self.pipe_initialized = False
         self.inpaint_pipe_initialized = False
         self.depth_pipe_initialized = False
@@ -35,7 +39,7 @@ def __init__(self, compute: Computing):
         self.controlnet = ControlnetContext(compute)
         self.stable_diffusion_xl = StableDiffusionXl(compute)
         self.app_settings = AppSettings().get_settings()
-        self.model_id = self.app_settings.model_settings.model_id
+        self.model_id = model_id
         self.low_vram_mode = self.app_settings.low_memory_mode
         self.wuerstchen = Wuerstchen(compute)