Update app.py

app.py

@@ -20,6 +20,11 @@ pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
+MODELS = [
+    "stabilityai/sdxl-turbo",
+    "CompVis/stable-diffusion-v1-4"
+]
+
 
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
@@ -68,6 +73,16 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # Text-to-Image Gradio Template")
 
+
+        with gr.Row():
+            model_dropdown = gr.Dropdown(
+                choices=list(MODELS.keys()),
+                value=DEFAULT_MODEL,
+                label="Select Model",
+                info="Choose which diffusion model to use"
+            )
+
+            
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",