Update app.py

app.py

@@ -6,15 +6,24 @@ import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
 
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
+
 MODEL_OPTIONS = [
     ("stabilityai/sdxl-turbo", "SDXL Turbo (Быстро)"),
     ("hakurei/waifu-diffusion", "Что-то альтернативное"),
 ]
 DEFAULT_MODEL_ID = "stabilityai/sdxl-turbo"
-# model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
 
-PIPELINES = {}
+if torch.cuda.is_available():
+    torch_dtype = torch.float16
+else:
+    torch_dtype = torch.float32
+
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 1024
+
+PIPELINES = {} 
 
 def load_pipelines():
     # SDXL Turbo
@@ -29,17 +38,8 @@ def load_pipelines():
     pipe = pipe.to(device)
     PIPELINES[mid] = pipe
 
-
-if torch.cuda.is_available():
-    torch_dtype = torch.float16
-else:
-    torch_dtype = torch.float32
-
 load_pipelines()
 
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 1024
-
 
 @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
@@ -90,7 +90,12 @@ css = """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # Text-to-Image Gradio Template")
-
+        
+        model_id = gr.Dropdown(
+            choices=[m[0] for m in MODEL_OPTIONS],
+            label="Model",
+            value=DEFAULT_MODEL_ID,)
+        
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",
@@ -109,7 +114,8 @@ with gr.Blocks(css=css) as demo:
                 label="Negative prompt",
                 max_lines=1,
                 placeholder="Enter a negative prompt",
-                visible=False,
+                visible=True,
+                value="dog, cat"
             )
 
             seed = gr.Slider(
@@ -161,6 +167,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn=infer,
         inputs=[
+            model_id,
             prompt,
             negative_prompt,
             seed,
@@ -174,4 +181,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()