Update app.py

app.py

@@ -26,10 +26,11 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float1
 
 
 def infer(prompt,samples):
+    all_images = [] 
     with context(device):
-        images = pipe(samples*[prompt], guidance_scale=7.5).images
-    return images
-
+        images = pipe(samples*[prompt],num_inference_steps=50, guidance_scale=7.5, height=400, width=400).sample
+        all_images.extend(images)
+    return all_images
 
 
 demo = gr.Blocks()