refactor gui.py

app.py

@@ -206,10 +206,11 @@ logger.setLevel(logging.DEBUG)
 
 
 # init GuiSD
-sd_gen = GuiSD(
+sd_gen: object = GuiSD(
     model_list=model_list,
     task_stablepy=task_stablepy,
-    lora_model_list=lora_model_list
+    lora_model_list=lora_model_list,
+    embed_list=embed_list,
 )
 
 with open("app.css", "r") as f:

gui.py

@@ -17,11 +17,16 @@ import logging
 from utils.string_utils import extract_parameters
 from stablepy import logger
 
+from models.upscaler import upscaler_dict_gui
+
 logging.getLogger("diffusers").setLevel(logging.ERROR)
 import diffusers
 
 diffusers.utils.logging.set_verbosity(40)
 import warnings
+from utils.download_utils import download_things
+
+hf_token: str = os.environ.get("HF_TOKEN")
 
 
 class GuiSD:
@@ -29,6 +34,7 @@ class GuiSD:
                  model_list,
                  task_stablepy,
                  lora_model_list,
+                 embed_list,
                  stream=True):
         self.model = None
 
@@ -43,6 +49,7 @@ class GuiSD:
         self.model_list = model_list
         self.task_stablepy = task_stablepy
         self.lora_model_list = lora_model_list
+        self.embed_list = embed_list
         self.stream = stream
 
     def load_new_model(
@@ -359,7 +366,7 @@ class GuiSD:
             "lora_scale_D": lora_scale4,
             "lora_E": lora5 if lora5 != "None" else None,
             "lora_scale_E": lora_scale5,
-            "textual_inversion": embed_list if textual_inversion and self.model.class_name != "StableDiffusionXLPipeline" else [],
+            "textual_inversion": self.embed_list if textual_inversion and self.model.class_name != "StableDiffusionXLPipeline" else [],
             "syntax_weights": syntax_weights,  # "Classic"
             "sampler": sampler,
             "xformers_memory_efficient_attention": xformers_memory_efficient_attention,