Update app.py

app.py

@@ -11,8 +11,15 @@ pipe.load_lora_weights(
 )
 pipe.to('cuda')
 @spaces.GPU
-def generate(prompt, negative_prompt, num_inference_steps, width, height):
-    return pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=num_inference_steps, width=width, height=height).images
+def generate(prompt, negative_prompt, num_inference_steps, width, height, num_samples):
+    return pipe(
+        prompt,
+        negative_prompt=negative_prompt,
+        num_inference_steps=num_inference_steps,
+        width=width,
+        height=height,
+        num_images_per_prompt=num_samples
+    ).images
 
 
 app = gr.Interface(
@@ -20,9 +27,10 @@ app = gr.Interface(
     inputs=[
         gr.Text(label="Prompt"),
         gr.Text("", label="Negative Prompt"),
-        gr.Number(45, label="Number inference steps"),
-        gr.Number(1024, label='image width'),
-        gr.Number(1024, label='image height'),
+        gr.Slider(min=1, max=100, default=45, label="Number inference steps"),
+        gr.Number(512, label='Image width'),
+        gr.Number(512, label='Image height'),
+        gr.Slider(min=1, max=10, default=1, label='Number samples'),
     ],
     outputs=gr.Gallery(),
 )