first

app.py

@@ -23,6 +23,7 @@ MAX_IMAGE_SIZE = 1024
 
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
+    model_id,
     prompt,
     negative_prompt,
     seed,
@@ -37,6 +38,9 @@ def infer(
         seed = random.randint(0, MAX_SEED)
 
     generator = torch.Generator().manual_seed(seed)
+    
+    pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
+    pipe = pipe.to(device)
 
     image = pipe(
         prompt=prompt,
@@ -67,7 +71,16 @@ css = """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # Text-to-Image Gradio Template")
-
+        
+        model_id_input = gr.Text(
+                label="Enter Model ID",
+                value="CompVis/stable-diffusion-v1-4",  
+                show_label=True,
+                placeholder="Enter model",
+            )
+        
+        
+        
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",
@@ -94,7 +107,7 @@ with gr.Blocks(css=css) as demo:
                 minimum=0,
                 maximum=MAX_SEED,
                 step=1,
-                value=0,
+                value=42,
             )
 
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
@@ -122,7 +135,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=0.0,
                     maximum=10.0,
                     step=0.1,
-                    value=0.0,  # Replace with defaults that work for your model
+                    value=7.0,  # Replace with defaults that work for your model
                 )
 
                 num_inference_steps = gr.Slider(
@@ -130,7 +143,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=1,
                     maximum=50,
                     step=1,
-                    value=2,  # Replace with defaults that work for your model
+                    value=20,  # Replace with defaults that work for your model
                 )
 
         gr.Examples(examples=examples, inputs=[prompt])
@@ -138,6 +151,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn=infer,
         inputs=[
+            model_id_input,
             prompt,
             negative_prompt,
             seed,