Update app.py

app.py

@@ -74,7 +74,7 @@ def init_predictor():
         transformer=transformer,
         text_encoder=text_encoder,
         torch_dtype=torch.bfloat16,
-    ).to("cpu")
+    ) #.to("cpu")
     
 negative_prompt = "Aerial view, aerial view, overexposed, low quality, deformation, a poor composition, bad hands, bad teeth, bad eyes, bad limbs, distortion"