fixed model change

app.py

@@ -31,6 +31,7 @@ MAX_IMAGE_SIZE = 1024
 
 @ZeroGPU
 def infer(
+    model_name,  
     prompt,
     negative_prompt,
     seed,
@@ -41,6 +42,17 @@ def infer(
     num_inference_steps,
     progress=gr.Progress(track_tqdm=True),
 ):
+
+    global pipe, current_model_id
+
+    # reload pipeline if user picked a different model
+    if model_name != current_model_id:
+        pipe = DiffusionPipeline.from_pretrained(
+            model_name,
+            torch_dtype=torch_dtype
+        ).to(device)
+        current_model_id = model_name
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)