Update app.py

app.py

@@ -23,6 +23,7 @@ MAX_IMAGE_SIZE = 1024
 
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
+    model,
     prompt,
     negative_prompt,
     seed,
@@ -33,7 +34,10 @@ def infer(
     progress=gr.Progress(track_tqdm=True),
 ):
     
-
+    global model_repo_id
+    if model != model_repo_id:
+        pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch_dtype)
+        pipe = pipe.to(device)
     generator = torch.Generator().manual_seed(seed)
 
     image = pipe(
@@ -61,10 +65,22 @@ css = """
     max-width: 640px;
 }
 """
-
+available_models = [
+    "CompVis/stable-diffusion-v1-4",
+    "stabilityai/sdxl-turbo"
+    "runwayml/stable-diffusion-v1-5",
+    "stabilityai/stable-diffusion-2-1",
+    "prompthero/openjourney",
+    
+]
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # Text-to-Image Gradio Template")
+        model = gr.Dropdown(
+            label="Model Selection",
+            choices=available_models,
+            value="CompVis/stable-diffusion-v1-4",
+        )
         prompt = gr.Text(
             label="Prompt",
             show_label=False,