Qafig commited on
Commit
5231cb0
·
verified ·
1 Parent(s): 14c73c0

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. script.py +1 -1
script.py CHANGED
@@ -129,7 +129,7 @@ def predict_video(bytes, path, n=30):
129
  all_probs = infer_imgs(frames)
130
  final_probs = sum(all_probs)/len(all_probs)
131
  os.remove(local_path)
132
- return 'real' if final_probs < 0.53 else 'generated', final_probs
133
 
134
  test = False
135
  # if 'qafigs_machine' in os.environ:
 
129
  all_probs = infer_imgs(frames)
130
  final_probs = sum(all_probs)/len(all_probs)
131
  os.remove(local_path)
132
+ return 'real' if final_probs < 0.47 else 'generated', final_probs
133
 
134
  test = False
135
  # if 'qafigs_machine' in os.environ: