Update app.py

app.py

@@ -24,10 +24,11 @@ def create_img(prompt :str, number_output_requested: int) -> list:
                                                         revision="fp16", 
                                                         torch_dtype=torch.float16, 
                                                         use_auth_token=AUTH_TOKEN)
-  generator.to("cpu")
+  generator.to("cuda")
   prompt = [prompt] * number_output_requested
-  images = generator(prompt).images
-  output_paths = save_images(images)
+  with torch.autocast("cuda"):
+    images = generator(prompt).images
+    output_paths = save_images(images)
   return output_paths
 
 diffusers_app = gr.Interface(
@@ -42,4 +43,4 @@ diffusers_app = gr.Interface(
         description="This is a basic app to generate pictures with Stable Diffusion."
 ) 
 
-diffusers_app.launch(debug=True)
+diffusers_app.launch()