Update app.py

app.py

@@ -24,11 +24,10 @@ def create_img(prompt :str, number_output_requested: int) -> list:
                                                         revision="fp16", 
                                                         torch_dtype=torch.float16, 
                                                         use_auth_token=AUTH_TOKEN)
-  
+  generator.to("cpu")
   prompt = [prompt] * number_output_requested
-  with torch.autocast("cpu"):
-    images = generator(prompt).images
-    output_paths = save_images(images)
+  images = generator(prompt).images
+  output_paths = save_images(images)
   return output_paths
 
 diffusers_app = gr.Interface(