Update app.py

app.py

@@ -40,7 +40,7 @@ def generate_image_fn(prompt: str, unconditional_guidance_scale: int) -> list:
         prompt,
         batch_size=num_images_to_gen,
         unconditional_guidance_scale=unconditional_guidance_scale,
-    ).to('cpu')
+    )
     end_time = time.time()
     print(f"Time taken: {end_time - start_time} seconds.")
     return [image for image in images]