Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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)
|