vumichien commited on
Commit
89327ab
·
1 Parent(s): 22d154c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,7 +39,7 @@ from huggingface_hub import hf_hub_download
39
  import gradio as gr
40
 
41
  ckpt_path = hf_hub_download('vumichien/AV-HuBERT', 'model.pt')
42
- user_dir = "/home/user/app/av_hubert/avhubert"
43
  face_detector_path = "/home/user/app/mmod_human_face_detector.dat"
44
  face_predictor_path = "/home/user/app/shape_predictor_68_face_landmarks.dat"
45
  mean_face_path = "/home/user/app/20words_mean_face.npy"
@@ -91,7 +91,7 @@ def predict(process_video):
91
  fo.write("".join(tsv_cont))
92
  with open(f"{data_dir}/test.wrd", "w") as fo:
93
  fo.write("".join(label_cont))
94
- # utils.import_user_module(Namespace(user_dir=user_dir))
95
  modalities = ["video"]
96
  gen_subset = "test"
97
  gen_cfg = GenerationConfig(beam=20)
 
39
  import gradio as gr
40
 
41
  ckpt_path = hf_hub_download('vumichien/AV-HuBERT', 'model.pt')
42
+ user_dir = "app/av_hubert/avhubert"
43
  face_detector_path = "/home/user/app/mmod_human_face_detector.dat"
44
  face_predictor_path = "/home/user/app/shape_predictor_68_face_landmarks.dat"
45
  mean_face_path = "/home/user/app/20words_mean_face.npy"
 
91
  fo.write("".join(tsv_cont))
92
  with open(f"{data_dir}/test.wrd", "w") as fo:
93
  fo.write("".join(label_cont))
94
+ utils.import_user_module(Namespace(user_dir=user_dir))
95
  modalities = ["video"]
96
  gen_subset = "test"
97
  gen_cfg = GenerationConfig(beam=20)