feat: limit

app.py

@@ -8,7 +8,7 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
 
 def inference(prompt, num_samples):
     all_images = [] 
-    images = pipe(prompt, num_images_per_prompt=num_samples, num_inference_steps=50, guidance_scale=7.5).images
+    images = pipe(prompt,height=512, width=768, num_images_per_prompt=num_samples, num_inference_steps=70, guidance_scale=7.5).images
     all_images.extend(images)
     return all_images
 
@@ -17,7 +17,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(label="prompt")
-            samples = gr.Slider(label="Samples",value=1)
+            samples = gr.Slider(label="Samples",value=1, step=1, maximum=6)
             run = gr.Button(value="Run")
         with gr.Column():
             gallery = gr.Gallery(show_label=False)