Update app.py

app.py

@@ -7,14 +7,14 @@ from diffusers import DiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
+# model_id = "stabilityai/sdxl-turbo"
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16
 else:
     torch_dtype = torch.float32
 
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -23,6 +23,7 @@ MAX_IMAGE_SIZE = 1024
 
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
+    model_id = "CompVis/stable-diffusion-v1-4",
     prompt,
     negative_prompt,
     seed,
@@ -39,6 +40,7 @@ def infer(
     generator = torch.Generator().manual_seed(seed)
 
     image = pipe(
+        model_id = model_id
         prompt=prompt,
         negative_prompt=negative_prompt,
         guidance_scale=guidance_scale,
@@ -69,6 +71,10 @@ with gr.Blocks(css=css) as demo:
         gr.Markdown(" # Text-to-Image Gradio Template")
 
         with gr.Row():
+            model_id = gr.Dropdown(
+            ["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4"], label="Model", info="Select the model you would like to use"
+            )
+            
             prompt = gr.Text(
                 label="Prompt",
                 show_label=False,