| import gradio as gr | |
| import inference_2 as inference | |
| title="Multimodal deepfake detector" | |
| description="Deepfake detection for videos, images and audio modalities." | |
| video_interface = gr.Interface(inference.deepfakes_video_predict, | |
| gr.Video(), | |
| "text", | |
| examples = ["videos/celeb_synthesis.mp4", "videos/real-1.mp4"], | |
| cache_examples = False | |
| ) | |
| image_interface = gr.Interface(inference.deepfakes_image_predict, | |
| gr.Image(), | |
| "text", | |
| examples = ["images/lady.jpg", "images/fake_image.jpg"], | |
| cache_examples=False | |
| ) | |
| audio_interface = gr.Interface(inference.deepfakes_spec_predict, | |
| gr.Audio(), | |
| "text", | |
| examples = ["audios/DF_E_2000027.flac", "audios/DF_E_2000031.flac"], | |
| cache_examples = False) | |
| app = gr.TabbedInterface(interface_list= [video_interface, audio_interface,image_interface], | |
| tab_names = ['Video inference', 'Audio inference', 'Image inference']) | |
| if __name__ == '__main__': | |
| app.launch(share = False) |