Move rent time from constant to gradio

app.py

@@ -31,7 +31,7 @@ from transformers import WhisperModel
 
 import spaces
 
-SPACES_MAX_DURATION = 5 * 60 # 5 minutes
+DEFAULT_RENT_TIME = 300
 
 if os.getenv('SYSTEM') == 'spaces':
     subprocess.run(["bash", "install.sh"])
@@ -189,11 +189,14 @@ def fast_check_ffmpeg():
     except:
         return False
 
+def get_duration():
+    step_duration = 3.75
+    return steps * step_duration
 
-@spaces.GPU(duration=SPACES_MAX_DURATION)
+@spaces.GPU(duration=get_duration)
 @torch.no_grad()
 def inference(audio_path, video_path, bbox_shift, extra_margin=10, parsing_mode="jaw", 
-              left_cheek_width=90, right_cheek_width=90, progress=gr.Progress(track_tqdm=True)):
+              left_cheek_width=90, right_cheek_width=90, rent_time=DEFAULT_RENT_TIME, progress=gr.Progress(track_tqdm=True)):
     # Set default parameters, aligned with inference.py
     args_dict = {
         "result_dir": './results/output', 
@@ -519,6 +522,7 @@ with gr.Blocks(css=css) as demo:
             parsing_mode = gr.Radio(label="Parsing Mode", choices=["jaw", "raw"], value="jaw")
             left_cheek_width = gr.Slider(label="Left Cheek Width", minimum=20, maximum=160, value=90, step=5)
             right_cheek_width = gr.Slider(label="Right Cheek Width", minimum=20, maximum=160, value=90, step=5)
+            rent_time = gr.Slider(label="Available/maximum rent time", minimum=1, maximum=1500, value=DEFAULT_RENT_TIME, step=1)
             bbox_shift_scale = gr.Textbox(label="'left_cheek_width' and 'right_cheek_width' parameters determine the range of left and right cheeks editing when parsing model is 'jaw'. The 'extra_margin' parameter determines the movement range of the jaw. Users can freely adjust these three parameters to obtain better inpainting results.")
 
             with gr.Row():
@@ -541,7 +545,8 @@ with gr.Blocks(css=css) as demo:
             extra_margin,
             parsing_mode,
             left_cheek_width,
-            right_cheek_width
+            right_cheek_width,
+            rent_time
         ],
         outputs=[out1,bbox_shift_scale]
     )