Give ZeroGPU longer to finish

app.py

@@ -39,11 +39,12 @@ TORCH_DTYPE = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 
 CONTROLNET = FluxControlNetModel.from_pretrained(CONTROLNET_MODEL, torch_dtype=TORCH_DTYPE)
 PIPE = FluxControlNetPipeline.from_pretrained(MODEL_DIR, controlnet=CONTROLNET, torch_dtype=TORCH_DTYPE)
+torch.cuda.empty_cache()
 PIPE = PIPE.to(DEVICE)
 
 MAX_SEED = np.iinfo(np.int32).max
 
-@spaces.GPU
+@spaces.GPU(duration=140)
 def infer(
     prompt,
     control_image_path,
@@ -84,10 +85,11 @@ with gr.Blocks(css=CSS) as demo:
                 type="filepath",
                 label="Control Image (LineArt)"
         )
-        prompt = gr.Textbox(
+        prompt = gr.Text(
             label="Prompt",
             placeholder="Enter your prompt",
             max_lines=1,
+            container=False
         )
         run_button = gr.Button("Generate", variant="primary")
         result = gr.Image(label="Result", show_label=False)
@@ -130,9 +132,10 @@ with gr.Blocks(css=CSS) as demo:
             ],
             inputs=[prompt]
         )
-
-    run_button.click(
-        infer,
+        
+    gr.on(
+        triggers=[run_button.click, prompt.submit],
+        fn = infer,
         inputs=[
             prompt,
             control_image,
@@ -142,7 +145,7 @@ with gr.Blocks(css=CSS) as demo:
             seed,
             randomize_seed
         ],
-        outputs=[result, seed]
+        outputs = [result, seed]
     )
 
 if __name__ == "__main__":