Update app.py

app.py

@@ -16,8 +16,8 @@ os.system('pip install gradio==3.12')
 os.system('pip install numpy==1.23.3')
 
 
-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')
+sys.path.append('/home/user/app/av_hubert/avhubert')
 
 print(sys.path)
 print(os.listdir())
@@ -31,19 +31,19 @@ import numpy as np
 import skvideo
 import skvideo.io
 from tqdm import tqdm
-from av_hubert.avhubert.preparation.align_mouth import landmarks_interpolate, crop_patch, write_video_ffmpeg
+from 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 av_hubert.fairseq
-from av_hubert.fairseq import checkpoint_utils, options, tasks, utils
-from av_hubert.fairseq.dataclass.configs import GenerationConfig
+import fairseq
+from fairseq import checkpoint_utils, options, tasks, utils
+from fairseq.dataclass.configs import GenerationConfig
 from huggingface_hub import hf_hub_download
 import gradio as gr
 
-os.chdir('/home/user/app/av_hubert/avhubert')
+# 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))