Hamidreza-Hashemp commited on
Commit
bde4a7b
·
1 Parent(s): c452162

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,7 +56,7 @@ def deepfakeclassifier(potential_test_video, option):
56
  return "This video is FAKE with {} probability!".format(predictions[0])
57
 
58
  demo = gr.Interface(fn=deepfakeclassifier, inputs=[gr.Video(),
59
- gr.Radio(["Original", "Custom"])] ,outputs="text", description="Original option uses the trained weights of the winning idea. Custom is my trained \
60
- network. Original optional performs better as it uses much more data for training!")
61
 
62
  demo.launch(debug=True)
 
56
  return "This video is FAKE with {} probability!".format(predictions[0])
57
 
58
  demo = gr.Interface(fn=deepfakeclassifier, inputs=[gr.Video(),
59
+ gr.Radio(["Pretrained", "Scratch"])] ,outputs="text", description="Pretrained option is training over the winning idea. Scratch is my training from \
60
+ the scratch. Pretrained optional performs better as it is trained with much more data for training!")
61
 
62
  demo.launch(debug=True)