Update app.py

app.py

@@ -14,10 +14,10 @@ os.system('pip install scikit-video')
 os.system('pip install transformers')
 os.system('pip install gradio==3.12')
 os.system('pip install numpy==1.23.3')
-os.chdir('/home/user/app/av_hubert/avhubert')
+
 
 sys.path.append('/home/user/app/av_hubert')
-sys.path.append('/home/user/app/av_hubert/avhubert')
+# sys.path.append('/home/user/app/av_hubert/avhubert')
 
 print(sys.path)
 print(os.listdir())
@@ -31,18 +31,20 @@ import numpy as np
 import skvideo
 import skvideo.io
 from tqdm import tqdm
-from preparation.align_mouth import landmarks_interpolate, crop_patch, write_video_ffmpeg
+from av_hubert.avhubert.preparation.align_mouth import landmarks_interpolate, crop_patch, write_video_ffmpeg
 from base64 import b64encode
 import torch
 import cv2
 import tempfile
 from argparse import Namespace
-import fairseq
-from fairseq import checkpoint_utils, options, tasks, utils
-from fairseq.dataclass.configs import GenerationConfig
+import av_hubert.fairseq
+from av_hubert.fairseq import checkpoint_utils, options, tasks, utils
+from av_hubert.fairseq.dataclass.configs import GenerationConfig
 from huggingface_hub import hf_hub_download
 import gradio as gr
 
+os.chdir('/home/user/app/av_hubert/avhubert')
+
 user_dir = "/home/user/app/av_hubert/avhubert"
 utils.import_user_module(Namespace(user_dir=user_dir))
 data_dir = tempfile.mkdtemp()