Update app.py

app.py

@@ -4,7 +4,7 @@ import subprocess
 from PIL import Image
 from io import BytesIO
 
-# Function to run the provided script
+# Function to run the provided script and return the path of the generated video
 def run_script(image_path, video_path):
     script = """
     !git clone https://github.com/s0md3v/roop.git
@@ -18,6 +18,9 @@ def run_script(image_path, video_path):
     """.format(video_path, image_path)
     
     subprocess.run(script, shell=True)
+    
+    # Return the path of the generated video
+    return "/content/swapped.mp4"
 
 # Streamlit app
 def main():
@@ -40,11 +43,11 @@ def main():
             with open(video_path, "wb") as f:
                 f.write(video_file.read())
 
-            # Run the script
-            run_script(image_path, video_path)
+            # Run the script and get the path of the generated video
+            generated_video_path = run_script(image_path, video_path)
 
             # Display the swapped video
-            video_bytes = open("/content/swapped.mp4", "rb").read()
+            video_bytes = open(generated_video_path, "rb").read()
             st.video(video_bytes)
 
 if __name__ == "__main__":