Update app.py

app.py

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