Update app.py

app.py

@@ -15,6 +15,9 @@ import os
 import ffmpeg
 from io import BytesIO
 import requests
+import sys
+
+python_path = sys.executable
 
 from fairseq.checkpoint_utils import load_model_ensemble_and_task_from_hf_hub
 from fairseq.models.text_to_speech.hub_interface import TTSHubInterface
@@ -108,7 +111,7 @@ def calculate(image_in, audio_in):
         f.write(jq_run.stdout.decode('utf-8').strip())
     # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     # os.system(f"rm -rf /content/image_audio.mp4")
-    os.system(f"cd /content/one-shot-talking-face && python3 -B test_script.py --img_path /content/image.png --audio_path /content/audio.wav --phoneme_path /content/test.json --save_dir /content/train")
+    os.system(f"cd /content/one-shot-talking-face && {python_path} -B test_script.py --img_path /content/image.png --audio_path /content/audio.wav --phoneme_path /content/test.json --save_dir /content/train")
     return "/content/train/image_audio.mp4"
 
 def merge_frames():
@@ -157,7 +160,7 @@ def one_shot_talking(image_in,audio_in):
     os.makedirs("/content/results")
   
   #Improve quality of input image
-  os.system(f"python /content/GFPGAN/inference_gfpgan.py --upscale 2 -i /content/image_pre.png -o /content/results --bg_upsampler realesrgan")
+  os.system(f"{python_path} /content/GFPGAN/inference_gfpgan.py --upscale 2 -i /content/image_pre.png -o /content/results --bg_upsampler realesrgan")
   # time.sleep(60)
   
   image_in_one_shot='/content/results/image_pre.png'
@@ -168,12 +171,12 @@ def one_shot_talking(image_in,audio_in):
   #Video Quality Improvement
   os.system(f"rm -rf /content/extracted_frames/image_audio_frames")
   #1. Extract the frames from the video file using PyVideoFramesExtractor
-  os.system(f"python /content/PyVideoFramesExtractor/extract.py --video=/content/train/image_audio.mp4")
+  os.system(f"{python_path} /content/PyVideoFramesExtractor/extract.py --video=/content/train/image_audio.mp4")
 
   #2. Improve image quality using GFPGAN on each frames
   # os.system(f"rm -rf /content/extracted_frames/image_audio_frames")
   os.system(f"rm -rf /content/video_results/")
-  os.system(f"python /content/GFPGAN/inference_gfpgan.py --upscale 2 -i /content/extracted_frames/image_audio_frames -o /content/video_results  --bg_upsampler realesrgan")
+  os.system(f"{python_path} /content/GFPGAN/inference_gfpgan.py --upscale 2 -i /content/extracted_frames/image_audio_frames -o /content/video_results  --bg_upsampler realesrgan")
 
   #3. Merge all the frames to a one video using imageio
   merge_frames()