Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 from refacer import Refacer
 import os
 import requests
+import tempfile  # Importing tempfile module
 
 # Hugging Face URL to download the model
 model_url = "https://huggingface.co/ofter/4x-UltraSharp/resolve/main/inswapper_128.onnx"
@@ -47,15 +48,19 @@ def run(video_path, *vars):
     refaced_video_path = refacer.reface(video_path, faces)
     print(f"Refaced video can be found at {refaced_video_path}")
     
-    # Convert the output video to memory buffer
-    video_buffer = io.BytesIO()
+    # Open the refaced video file and write to a temporary file
     with open(refaced_video_path, "rb") as f:
-        video_buffer.write(f.read())
-    
-    # Rewind the buffer to the beginning
-    video_buffer.seek(0)
-    
-    return video_buffer  # Gradio will handle the video display
+        # Use tempfile to store the refaced video in memory
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+        temp_file.write(f.read())
+        temp_file.close()  # Close the temporary file
+
+        # Reopen the temp file to use in the Gradio UI
+        with open(temp_file.name, "rb") as temp_f:
+            video_buffer = io.BytesIO(temp_f.read())
+        
+        # Return the video buffer for Gradio to display in the UI
+        return video_buffer  # Gradio will handle the video display
 
 # Prepare Gradio components
 with gr.Blocks() as demo: