Update app.py

app.py

@@ -26,7 +26,7 @@ def create_img(prompt :str, number_output_requested: int) -> list:
                                                         use_auth_token=AUTH_TOKEN)
   
   prompt = [prompt] * number_output_requested
-  with torch.autocast():
+  with torch.autocast("cpu"):
     images = generator(prompt).images
     output_paths = save_images(images)
   return output_paths