Kumar commited on
Commit
4d1de43
·
1 Parent(s): 297ee2e

update output labels

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -38,7 +38,10 @@ gr.Interface(
38
  inputs=[gr.Video(label="Upload a Video"),
39
  gr.Dropdown(choices=conv_layer_names, label="Model target layer", value=conv_layer_names[-1])
40
  ],
41
- outputs=["text", "image"],
 
 
 
42
  title="Deepfake video detection system",
43
  description="Upload a video, and the model will predict whether its a deepfake content, with RAI features."
44
  ).launch()
 
38
  inputs=[gr.Video(label="Upload a Video"),
39
  gr.Dropdown(choices=conv_layer_names, label="Model target layer", value=conv_layer_names[-1])
40
  ],
41
+ outputs=[
42
+ gr.Textbox(label="Prediction"),
43
+ gr.Image(label="Grad-CAM Visualization")
44
+ ],
45
  title="Deepfake video detection system",
46
  description="Upload a video, and the model will predict whether its a deepfake content, with RAI features."
47
  ).launch()