Update app.py

app.py

@@ -17,8 +17,7 @@ MAX_IMAGE_SIZE = 2048
 pipe = GlmImagePipeline.from_pretrained(
     "zai-org/GLM-Image",
     torch_dtype=torch.bfloat16,
-    device_map="cuda"
-)
+).to("cuda")
 
 
 @spaces.GPU(duration=120)