Update app.py

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)