Update app.py

app.py

@@ -6,6 +6,7 @@ from io import BytesIO
 
 # Function to run the provided script and return the path of the generated video
 def run_script(image_path, video_path):
+    output_video_path = "/content/swapped.mp4"
     script = """
     !git clone https://github.com/s0md3v/roop.git
     %cd roop
@@ -14,13 +15,13 @@ def run_script(image_path, video_path):
     !mkdir models
     !mv inswapper_128.onnx ./models
 
-    !python run.py --target {} --output-video-quality 80 --source {} -o /content/swapped.mp4 --execution-provider cuda --frame-processor face_swapper face_enhancer
-    """.format(video_path, image_path)
+    !python run.py --target {} --output-video-quality 80 --source {} -o {} --execution-provider cuda --frame-processor face_swapper face_enhancer
+    """.format(video_path, image_path, output_video_path)
     
     subprocess.run(script, shell=True)
     
     # Return the path of the generated video
-    return "/root/roop/content/swapped.mp4"
+    return output_video_path
 
 # Streamlit app
 def main():
@@ -34,8 +35,8 @@ def main():
     if st.sidebar.button("Swap Faces"):
         if image_file is not None and video_file is not None:
             # Save uploaded files
-            image_path = "./uploaded_image.jpg"
-            video_path = "./uploaded_video.mp4"
+            image_path = "/content/uploaded_image.jpg"
+            video_path = "/content/uploaded_video.mp4"
 
             with open(image_path, "wb") as f:
                 f.write(image_file.read())