Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import os
-from IPython.display import clear_output
 
 # Your Lipsync code here
 def main():
@@ -18,34 +17,35 @@ def main():
     if st.sidebar.button("Sync Lips"):
         if audio_file and video_file:
             # Save audio file
-            with open("/content/sample_data/input_audio.wav", "wb") as audio_writer:
+            audio_file_path = "input_audio.wav"
+            with open(audio_file_path, "wb") as audio_writer:
                 audio_writer.write(audio_file.read())
 
             # Save video file
-            with open("/content/sample_data/input_vid.mp4", "wb") as video_writer:
+            video_file_path = "input_vid.mp4"
+            with open(video_file_path, "wb") as video_writer:
                 video_writer.write(video_file.read())
 
             # Run Lipsync code
-            pad_top =  0
-            pad_bottom =  10
-            pad_left =  0
-            pad_right =  0
-            rescaleFactor =  1
+            pad_top = 0
+            pad_bottom = 10
+            pad_left = 0
+            pad_right = 0
+            rescale_factor = 1
             nosmooth = True
             use_hd_model = False
             checkpoint_path = 'checkpoints/wav2lip.pth' if not use_hd_model else 'checkpoints/wav2lip_gan.pth'
 
-            cmd = f"!python inference.py --checkpoint_path {checkpoint_path} --face '/content/sample_data/input_vid.mp4' --audio '/content/sample_data/input_audio.wav' --pads {pad_top} {pad_bottom} {pad_left} {pad_right} --resize_factor {rescaleFactor}"
+            cmd = f"python inference.py --checkpoint_path {checkpoint_path} --face '{video_file_path}' --audio '{audio_file_path}' --pads {pad_top} {pad_bottom} {pad_left} {pad_right} --resize_factor {rescale_factor}"
             if nosmooth:
                 cmd += " --nosmooth"
 
             os.system(cmd)
 
-            output_file_path = '/content/Wav2Lip/results/result_voice.mp4'
+            output_file_path = 'results/result_voice.mp4'
 
             # Display output video
             if os.path.exists(output_file_path):
-                clear_output()
                 st.video(output_file_path)
             else:
                 st.error("Processing failed. Output video not found.")