fps=24

app.py

@@ -55,7 +55,7 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 # ----------------------------------------
 # Step 3: Inference Function
 # ----------------------------------------
-def run_inference(video_path, save_format):
+def run_inference(video_path, save_format, upscale):
     input_name = f"{uuid.uuid4()}.mp4"
     input_path = os.path.join(UPLOAD_DIR, input_name)
     shutil.copy(video_path, input_path)
@@ -66,7 +66,8 @@ def run_inference(video_path, save_format):
         "--model_path", MODEL_PATH,
         "--output_path", OUTPUT_DIR,
         "--is_vae_st",
-        "--save_format", save_format
+        "--save_format", save_format,
+        "--upscale", str(upscale)  # ✅ add this
     ]
 
     try:
@@ -143,13 +144,14 @@ with gr.Blocks() as demo:
             value="yuv444p",
             label="Save format"
         )
+        upscale_input = gr.Number(value=4, label="Upscale Factor (e.g. 2, 4, 8)")
 
     run_button = gr.Button("Run Inference")
     status = gr.Textbox(label="Status")
 
     run_button.click(
         fn=run_inference,
-        inputs=[input_video, save_format],
+        inputs=[input_video, save_format, upscale_input],  # ✅ added upscale_input
         outputs=[status, output_video],
     )