Update app.py

app.py

@@ -12,7 +12,6 @@ SAFETY_CHECKER = False
 
 loaded_acc = None
 device = "cuda"
-#device = "cuda" if torch.cuda.is_available() else "cpu"
 
 ACC_lora={
     "TDD":"sdxl_tdd_wo_adv_lora.safetensors",
@@ -79,7 +78,7 @@ if SAFETY_CHECKER:
         return images, has_nsfw_concepts
 
 
-@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True, duration=3)
 def generate_image(
     prompt,
     negative_prompt,
@@ -192,7 +191,7 @@ Target-Driven Distillation (TDD) is a state-of-the-art consistency distillation
 
         with gr.Column(scale=1):
             with gr.Group():
-                img = gr.Image(label="TDD Image", value="/share/wangcunzheng/test1.png")
+                img = gr.Image(label="TDD Image", value="cat.png")
                 submit_sdxl = gr.Button("Run on SDXL")
     gr.Examples(
         examples=[