Upload app.py

app.py

@@ -192,13 +192,15 @@ class GuiSD:
     def load_new_model(self, model_name, vae_model, task, controlnet_model, progress=gr.Progress(track_tqdm=True)):
 
         # download link model > model_name
-
-        model_name = valid_model_name(model_name) # MOD
+        if "http" in model_name: #
+            model_name, model_type = download_link_model(model_name, DIRECTORY_MODELS) #
+            is_link_model = True #
+        else: is_link_model = False #
 
         self.update_storage_models()
 
         vae_model = vae_model if vae_model != "None" else None
-        model_type = get_model_type(model_name)
+        model_type = get_model_type(model_name) if not is_link_model else model_type #
         dtype_model = torch.bfloat16 if model_type == "FLUX" else torch.float16
 
         if not os.path.exists(model_name):