Hamidreza-Hashemp commited on
Commit
38a364a
1 Parent(s): bde4a7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ import gradio as gr
13
 
14
 
15
  def deepfakeclassifier(potential_test_video, option):
16
- if option == 'Original':
17
  weights_dir = "./weights"
18
  models_dir = ["Original_DeepFakeClassifier_tf_efficientnet_b7_ns"]
19
  else:
 
13
 
14
 
15
  def deepfakeclassifier(potential_test_video, option):
16
+ if option == 'Pretrained':
17
  weights_dir = "./weights"
18
  models_dir = ["Original_DeepFakeClassifier_tf_efficientnet_b7_ns"]
19
  else: