| import gradio as gr |
| import pipeline |
|
|
|
|
| title="EfficientNetV2 Deepfakes Video Detector" |
| description="EfficientNetV2 Deepfakes Image Detector by using frame-by-frame detection." |
| |
| |
| video_interface = gr.Interface(pipeline.deepfakes_video_predict, |
| gr.Video(), |
| "text", |
| examples = ["videos/celeb_synthesis.mp4", "videos/real-1.mp4"], |
| cache_examples = False |
| ) |
|
|
|
|
| image_interface = gr.Interface(pipeline.deepfakes_image_predict, |
| gr.Image(), |
| "text", |
| examples = ["images/lady.jpg", "images/fake_image.jpg"], |
| cache_examples=False |
| ) |
|
|
| audio_interface = gr.Interface(pipeline.deepfakes_audio_predict, |
| gr.Audio(), |
| "text", |
| examples = ["audios/DF_E_2000027.flac", "audios/DF_E_2000031.flac"], |
| cache_examples = False) |
|
|
|
|
| app = gr.TabbedInterface(interface_list= [image_interface, video_interface, audio_interface], |
| tab_names = ['Image inference', 'Video inference', 'Audio inference']) |
|
|
| if __name__ == '__main__': |
| app.launch(share = False) |