Update app.py

app.py

@@ -2,26 +2,24 @@ import gradio as gr
 import subprocess
 import os
 import spaces
+import inference_video_w
+import torch
 
 # Download the file
 subprocess.run([
     "wget",
-    "--no-check-certificate",
-    "https://drive.google.com/uc?id=1mj9lH6Be7ztYtHAr1xUUGT3hRtWJBy_5",
+    "https://huggingface.co/r3gm/RIFE/resolve/main/RIFEv4.26_0921.zip",
     "-O",
-    "RIFE_trained_model_v4.13.2.zip"
+    "RIFEv4.26_0921.zip"
 ], check=True)
 
 # Unzip the downloaded file
 subprocess.run([
     "unzip",
     "-o", 
-    "RIFE_trained_model_v4.13.2.zip"
+    "RIFEv4.26_0921.zip"
 ], check=True)
 
-# The name of your script
-SCRIPT_NAME = "inference_video.py"
-
 @spaces.GPU(duration=120)
 def run_rife(
     input_video, 
@@ -29,7 +27,6 @@ def run_rife(
     time_exponent, 
     fixed_fps, 
     video_scale, 
-    enable_uhd_mode, 
     remove_duplicate_frames, 
     create_montage,
     progress=gr.Progress(track_tqdm=True),
@@ -40,58 +37,37 @@ def run_rife(
     ext = "mp4"
     model_dir = "train_log"
     
+    # Construct output filename pattern to match what inference_video.py expects/generates
     video_path_wo_ext = os.path.splitext(os.path.basename(input_video))[0]
-    output_base_name = "{}_{}X_{}fps.{}".format(video_path_wo_ext, int(frame_multiplier), int(fixed_fps), ext)
-    
-    cmd = ["python3", SCRIPT_NAME]
-    cmd.extend(["--video", input_video])
-    cmd.extend(["--output", output_base_name])
-    cmd.extend(["--multi", str(int(frame_multiplier))])
+    # We pass the desired output name, though the function logic tries to stick to this pattern anyway
+    output_base_name = "{}_{}X_fps.{}".format(video_path_wo_ext, int(frame_multiplier), ext)
     
-    if time_exponent != 1:
-        cmd.extend(["--exp", str(int(time_exponent))])
-
     if fixed_fps > 0:
-        cmd.extend(["--fps", str(int(fixed_fps))])
         gr.Warning("Will not merge audio because using fps flag!")
-        
-    if video_scale != 1.0:
-        cmd.extend(["--scale", str(video_scale)])
-
-    cmd.extend(["--ext", ext])
-    cmd.extend(["--model", model_dir])
-    cmd.append("--fp16") 
 
-    if enable_uhd_mode:
-        cmd.append("--UHD")
-        
-    if remove_duplicate_frames:
-        cmd.append("--skip")
-        
-    if create_montage:
-        cmd.append("--montage")
-
-    print(f"Executing command: {' '.join(cmd)}")
+    print(f"Starting Inference for: {input_video}")
 
     try:
-        process = subprocess.run(cmd, capture_output=True, text=True)
-        
-        if process.stdout:
-            print("STDOUT:", process.stdout)
-        if process.stderr:
-            print("STDERR:", process.stderr)
-
-        if process.returncode != 0:
-            raise gr.Error(f"Inference failed. Error: {process.stderr}")
-
-        final_output_file = f"{output_base_name}.{ext}"
-        
-        if os.path.exists(final_output_file):
-            return final_output_file
+        # Call the imported function directly
+        result_path = inference_video_w.inference(
+            video=input_video,
+            output=output_base_name,
+            modelDir=model_dir,
+            fp16=(True if torch.cuda.is_available() else False),
+            UHD=False,
+            scale=video_scale,
+            skip=remove_duplicate_frames,
+            fps=int(fixed_fps) if fixed_fps > 0 else None,
+            ext=ext,
+            exp=int(time_exponent),
+            multi=int(frame_multiplier),
+            montage=create_montage
+        )
+
+        if result_path and os.path.exists(result_path):
+            return result_path
         else:
-            if os.path.exists(output_base_name):
-                return output_base_name
-            raise gr.Error(f"Output file not found. Expected: {final_output_file}")
+            raise gr.Error(f"Output file not found. Expected: {result_path}")
 
     except Exception as e:
         raise gr.Error(f"An error occurred: {str(e)}")
@@ -99,9 +75,9 @@ def run_rife(
 
 # --- Gradio UI Layout ---
 
-with gr.Blocks(title="Professional FPS Booster") as app:
-    gr.Markdown("# ⚡ RIFE: High-Performance FPS Booster")
-    gr.Markdown("Maximize video fluidity and frame rate using deep flow estimation.")
+with gr.Blocks(title="Frame Rate Enhancer") as app:
+    gr.Markdown("# ⚡ RIFE: Frame Rate Enhancer")
+    gr.Markdown("Creates extra frames between the original ones to make motion in your videos smoother and more fluid.")
     gr.Markdown("⚠️ **Notice:** Keep input videos under 60 seconds for frame interpolation to prevent GPU task aborts.")
 
     with gr.Row():
@@ -109,13 +85,11 @@ with gr.Blocks(title="Professional FPS Booster") as app:
         with gr.Column(scale=1):
             input_vid = gr.Video(label="🎬 Input Source Video", sources=["upload"])
             
-            with gr.Group():
-                gr.Markdown("### 🚀 Performance Settings")
-                
+            with gr.Group():                
                 multi_param = gr.Dropdown(
                     choices=["2", "3", "4", "5", "6"], 
                     value="2", 
-                    label="⚡ Frame Multiplier",
+                    label="🗃️ Frame Multiplier",
                     info="2X = Double FPS (e.g. 30 -> 60). Higher multipliers create more intermediate frames."
                 )
 
@@ -141,15 +115,10 @@ with gr.Blocks(title="Professional FPS Booster") as app:
                     )
 
                 with gr.Row():
-                    uhd_chk = gr.Checkbox(
-                        label="💎 UHD/4K Optimization", 
-                        value=False,
-                        info="Enable for high-resolution sources."
-                    )
                     skip_chk = gr.Checkbox(
                         label="⏩ Skip Static Frames", 
                         value=False,
-                        info="Bypass processing for non-moving scenes to save time."
+                        info="Bypass processing for static frames to save time."
                     )
                     montage_chk = gr.Checkbox(
                         label="🆚 Split-Screen Comparison", 
@@ -157,11 +126,11 @@ with gr.Blocks(title="Professional FPS Booster") as app:
                         info="Output video showing Original vs. Processed."
                     )
 
-            btn_run = gr.Button("🚀 RENDER HIGH FPS VIDEO", variant="primary", size="lg")
+            btn_run = gr.Button("GENERATE INTERMEDIATE FRAMES", variant="primary", size="lg")
 
         # --- Right Column: Output ---
         with gr.Column(scale=1):
-            output_vid = gr.Video(label="⚡ High FPS Result")
+            output_vid = gr.Video(label="INTERPOLATED RESULT")
             gr.Markdown("**Status:** Rendering time depends on input resolution and duration.")
 
     # --- Bind Logic ---
@@ -173,7 +142,6 @@ with gr.Blocks(title="Professional FPS Booster") as app:
             exp_param, 
             fps_param, 
             scale_param, 
-            uhd_chk, 
             skip_chk, 
             montage_chk
         ],