Update app.py

app.py

@@ -3,10 +3,11 @@ import gradio as gr
 from refacer import Refacer
 import os
 import requests
+from huggingface_hub import HfApi
 
 # 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
+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():
@@ -44,8 +45,14 @@ def run(*vars):
                 'threshold': thresholds[k]
             })
 
+    # Specify the output path for the refaced video
+    output_dir = "/home/user/app/out"
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+    refaced_video_path = os.path.join(output_dir, "refaced_video.mp4")
+
     # Call refacer to process video and get refaced video path
-    refaced_video_path = refacer.reface(video_path, faces)  # Get refaced video path
+    refacer.reface(video_path, faces, output_path=refaced_video_path)
     print(f"Refaced video can be found at {refaced_video_path}")
     
     # Convert the output video to memory buffer
@@ -85,5 +92,20 @@ with gr.Blocks() as demo:
     # Click event: Refacing the video and showing the refaced video in Gradio
     button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
 
+# Function to upload the refaced video to Hugging Face Spaces
+def upload_to_hf(video_path):
+    api = HfApi()
+    repo_id = "your-username/your-space-name"  # Replace with your actual repository name
+    api.upload_file(
+        path_or_fileobj=video_path,
+        path_in_repo="out/refaced_video.mp4",
+        repo_id=repo_id,
+        repo_type="space"
+    )
+    print("Refaced video uploaded to Hugging Face Spaces.")
+
+# Call the upload function after refacing is complete
+upload_to_hf(refaced_video_path)
+
 # Launch the Gradio app
 demo.queue().launch(show_error=True, server_name="0.0.0.0", server_port=7860)