Spaces:
Build error
Build error
Commit
·
b0df336
1
Parent(s):
96e5a5e
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ device = "CPU" if len(gpus) == 0 else "GPU"
|
|
20 |
print("using {}".format(device))
|
21 |
|
22 |
model_path = '3_ff_raw.pkl'
|
23 |
-
|
24 |
def predict(video):
|
25 |
path = extract_landmark(video)
|
26 |
test_samples, test_samples_diff, _, _, test_sv, test_vc = get_data_for_test(path, 1, block_size)
|
@@ -80,7 +80,7 @@ def predict(video):
|
|
80 |
label = "Real"
|
81 |
print("the pd is {}".format(pd))
|
82 |
|
83 |
-
output_video = test_full_image_network(video,model_path)
|
84 |
return label,output_video
|
85 |
|
86 |
inputs = gr.inputs.Video()
|
|
|
20 |
print("using {}".format(device))
|
21 |
|
22 |
model_path = '3_ff_raw.pkl'
|
23 |
+
output_path= 'output'
|
24 |
def predict(video):
|
25 |
path = extract_landmark(video)
|
26 |
test_samples, test_samples_diff, _, _, test_sv, test_vc = get_data_for_test(path, 1, block_size)
|
|
|
80 |
label = "Real"
|
81 |
print("the pd is {}".format(pd))
|
82 |
|
83 |
+
output_video = test_full_image_network(video,model_path,output_path)
|
84 |
return label,output_video
|
85 |
|
86 |
inputs = gr.inputs.Video()
|