Update app.py

app.py

@@ -21,6 +21,12 @@ sys.path.append('/home/user/app/av_hubert')
 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
 import numpy as np
 import skvideo
@@ -31,7 +37,7 @@ 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
@@ -39,7 +45,6 @@ from huggingface_hub import hf_hub_download
 import gradio as gr
 
 ckpt_path = hf_hub_download('vumichien/AV-HuBERT', 'model.pt')
-user_dir = "/home/user/app/av_hubert/avhubert"
 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"
@@ -91,7 +96,7 @@ def predict(process_video):
       fo.write("".join(tsv_cont))
     with open(f"{data_dir}/test.wrd", "w") as fo:
       fo.write("".join(label_cont))
-    utils.import_user_module(Namespace(user_dir=user_dir))
+    
     modalities = ["video"]
     gen_subset = "test"
     gen_cfg = GenerationConfig(beam=20)