Upload folder using huggingface_hub
Browse files
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.
|
| 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:
|