Update app.py

app.py

@@ -14,18 +14,8 @@ api = HfApi()
 import torch
 from diffusers import DiffusionPipeline, AutoencoderKL
 
-vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-
-pipe = DiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
-    vae=vae, torch_dtype=torch.float16, variant="fp16",
-    use_safetensors=True
-)
-
 device="cuda" if torch.cuda.is_available() else "cpu"
 
-pipe.to(device)
-
 def get_files(file_paths):
     last_files = {}  # Dictionary to store the last file for each path
 
@@ -72,7 +62,17 @@ def load_model(custom_model):
 
 
 def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
+    
+    vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+    
+    pipe = DiffusionPipeline.from_pretrained(
+        "stabilityai/stable-diffusion-xl-base-1.0",
+        vae=vae, torch_dtype=torch.float16, variant="fp16",
+        use_safetensors=True
+    )
 
+    pipe.to(device)
+    
     if weight_name == "NO SAFETENSORS FILE": 
         pipe.load_lora_weights(
             custom_model,