Update app.py

app.py

@@ -2,9 +2,30 @@ import gradio as gr
 from refacer import Refacer
 import argparse
 import os
+import requests
+import tempfile
+import shutil
 
-# Hugging Face Space-இல் model path
-model_path = "./models/inswapper_128.onnx"
+# Hugging Face URL to download the model
+model_url = "https://huggingface.co/ofter/4x-UltraSharp/resolve/main/inswapper_128.onnx"
+model_path = "/home/user/app/inswapper_128.onnx"  # absolute path for the model in your environment
+
+# Function to download the model if not exists
+def download_model():
+    if not os.path.exists(model_path):
+        print("Downloading the inswapper_128.onnx model...")
+        response = requests.get(model_url)
+        if response.status_code == 200:
+            with open(model_path, 'wb') as f:
+                f.write(response.content)
+            print("Model downloaded successfully!")
+        else:
+            print(f"Error: Model download failed. Status code: {response.status_code}")
+    else:
+        print("Model already exists.")
+
+# Download the model when the script runs
+download_model()
 
 # Argument parser
 parser = argparse.ArgumentParser(description='Refacer')
@@ -21,7 +42,7 @@ refacer = Refacer(force_cpu=args.force_cpu, colab_performance=args.colab_perform
 
 num_faces = args.max_num_faces
 
-# Run function for refacing video (without involving ffmpeg directly)
+# Run function for refacing video
 def run(*vars):
     video_path = vars[0]
     origins = vars[1:(num_faces+1)]
@@ -37,11 +58,11 @@ def run(*vars):
                 'threshold': thresholds[k]
             })
 
-    # Call refacer to process video (skip ffmpeg related parts)
-    refaced_video_path = refacer.reface(video_path, faces)  # refaced video path
+    # Call refacer to process video and get refaced video path
+    refaced_video_path = refacer.reface(video_path, faces)  # Get refaced video path
     print(f"Refaced video can be found at {refaced_video_path}")
 
-    # Instead of saving to disk, return the path to Gradio UI directly
+    # Directly return the path to the Gradio UI without using ffmpeg or temp files
     return refaced_video_path  # Gradio will handle the video display
 
 # Prepare Gradio components