Update app.py (#2)

- Update app.py (790e413a3fa5d3f4f021603040a3f65f2fc9f1c8)


Co-authored-by: Jevon Edmund <[email protected]>

app.py

@@ -26,9 +26,9 @@ def display_example_image(url):
 def generate_images_using_huggingface_diffusers(text):
 #    pipe = StableDiffusionPipeline.from_pretrained("sd-dreambooth-library/cat-toy", torch_dtype=torch.float16)
     pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float32)    
-    pipe = pipe.to("cpu")
+    pipe = pipe.to("cuda")
     prompt = text
-    image = pipe(prompt).images[0]
+    image = pipe(prompt,num_images_per_prompt=3)
     return image
  
 # Placeholder function for generating images (replace this with your actual generative AI code)