asdasdasdasd commited on
Commit
b0df336
·
1 Parent(s): 96e5a5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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()