add number of images

app.py

@@ -2,19 +2,24 @@ import gradio as gr
 from diffusers import DDPMPipeline
 
 model = 'alexktrs/CumulusCloudsGenerator'
-device='cpu'
+device='cuda'
 
 generator = DDPMPipeline.from_pretrained(model)
 generator.to(device)
 
-def generate(num_inference_steps):
+def generate(num_images, num_inference_steps):
+    images=[]
+    print(num_images)
+    if num_images==None:
+        num_images=1
+    num_images=int(num_images)
 
-    image1 = generator(num_inference_steps=num_inference_steps).images[0]
-    image2 = generator(num_inference_steps=num_inference_steps).images[0]
-    
-    return [image1, image2]
+    for i in range(num_images):
+        image = generator(num_inference_steps=num_inference_steps).images[0]
+        images.append(image)
+    return images
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
                 
                 # Generate Cumulus Clouds
@@ -22,9 +27,12 @@ with gr.Blocks() as demo:
                 """)
     
     gallery=gr.Gallery(type="pil")
-    slider=gr.Slider(label='Inference Steps', minimum=0, maximum=100)
-    
+
+    with gr.Row():
+        slider=gr.Slider(label='Inference Steps', minimum=1, maximum=100, step=1, value=20)
+        n=gr.Number(label='Number of Generated Images', minimum=1, maximum=4, value=2)
+
     btn = gr.Button("Generate Clouds")
-    btn.click(fn=generate, inputs=slider, outputs=gallery)
+    btn.click(fn=generate, inputs=[n, slider], outputs=gallery)
 
 demo.launch()