Update app.py

app.py

@@ -17,14 +17,15 @@ 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')
 
 print(sys.path)
 print(os.listdir())
 
 from fairseq import checkpoint_utils, options, tasks, utils
 from argparse import Namespace
-user_dir = "/home/user/app/av_hubert/avhubert"
-utils.import_user_module(Namespace(user_dir=user_dir))
+
+
 
 
 import dlib, cv2, os
@@ -44,11 +45,13 @@ from fairseq.dataclass.configs import GenerationConfig
 from huggingface_hub import hf_hub_download
 import gradio as gr
 
+user_dir = "/home/user/app/av_hubert/avhubert"
 ckpt_path = hf_hub_download('vumichien/AV-HuBERT', 'model.pt')
 face_detector_path = "/home/user/app/mmod_human_face_detector.dat"
 face_predictor_path = "/home/user/app/shape_predictor_68_face_landmarks.dat"
 mean_face_path = "/home/user/app/20words_mean_face.npy"
 mouth_roi_path = "/home/user/app/roi.mp4"
+utils.import_user_module(Namespace(user_dir=user_dir))
 
 def detect_landmark(image, detector, predictor):
     gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
@@ -96,7 +99,6 @@ def predict(process_video):
       fo.write("".join(tsv_cont))
     with open(f"{data_dir}/test.wrd", "w") as fo:
       fo.write("".join(label_cont))
-    
     modalities = ["video"]
     gen_subset = "test"
     gen_cfg = GenerationConfig(beam=20)