Update app.py

app.py

@@ -10,8 +10,8 @@ import PIL
 base = "stabilityai/stable-diffusion-xl-base-1.0"
 repo = "tianweiy/DMD2"
 checkpoints = {
-    "1-Step" : ["dmd2_sdxl_1step_unet_fp16.bin", 1],
-    "4-Step" : ["dmd2_sdxl_4step_unet_fp16.bin", 4],
+    "1-Step" : ["dmd2_sdxl_1step_unet.bin", 1],
+    "4-Step" : ["dmd2_sdxl_4step_unet.bin", 4],
 }
 loaded = None
 
@@ -28,7 +28,7 @@ if torch.cuda.is_available():
 
 
 # Function 
-@spaces.GPU(enable_queue=True)
+@spaces.GPU()
 def generate_image(prompt, ckpt):
     global loaded
     print(prompt, ckpt)