demo / app.py
ybbwcwaps
ui
9011f2d
raw
history blame
No virus
723 Bytes
import gradio as gr
from run import get_model, detect_video
model = get_model()
def greet(video):
print(video, type(video))
pred = detect_video(video_path=video, model=model)
if pred > 0.5:
string = f"Fake: {pred*100:.2f}%"
else:
string = f"Real: {(1-pred)*100:.2f}%"
print(string)
return string
with gr.Blocks() as demo:
gr.Markdown("# Fake Video Detector")
with gr.Tabs():
with gr.TabItem("Video Detect"):
with gr.Row():
video_input = gr.Video()
detect_output = gr.Textbox("Output")
video_button = gr.Button("detect")
video_button.click(greet, inputs=video_input, outputs=detect_output)
demo.launch()