vumichien commited on
Commit
22d154c
1 Parent(s): e5c36a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,6 +17,7 @@ os.system('pip install numpy==1.23.3')
17
  os.chdir('/home/user/app/av_hubert/avhubert')
18
 
19
  sys.path.append('/home/user/app/av_hubert')
 
20
 
21
  print(sys.path)
22
 
@@ -25,7 +26,7 @@ import numpy as np
25
  import skvideo
26
  import skvideo.io
27
  from tqdm import tqdm
28
- from avhubert.preparation.align_mouth import landmarks_interpolate, crop_patch, write_video_ffmpeg
29
  from base64 import b64encode
30
  import torch
31
  import cv2
@@ -90,7 +91,7 @@ def predict(process_video):
90
  fo.write("".join(tsv_cont))
91
  with open(f"{data_dir}/test.wrd", "w") as fo:
92
  fo.write("".join(label_cont))
93
- utils.import_user_module(Namespace(user_dir=user_dir))
94
  modalities = ["video"]
95
  gen_subset = "test"
96
  gen_cfg = GenerationConfig(beam=20)
 
17
  os.chdir('/home/user/app/av_hubert/avhubert')
18
 
19
  sys.path.append('/home/user/app/av_hubert')
20
+ sys.path.append('/home/user/app/av_hubert/avhubert')
21
 
22
  print(sys.path)
23
 
 
26
  import skvideo
27
  import skvideo.io
28
  from tqdm import tqdm
29
+ from preparation.align_mouth import landmarks_interpolate, crop_patch, write_video_ffmpeg
30
  from base64 import b64encode
31
  import torch
32
  import cv2
 
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)