23b719w commited on
Commit
ca61b7c
·
1 Parent(s): 76503f2

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -171,8 +171,8 @@ detection_model = load_model()
171
  # predicted_img.save('predicted.jpg')
172
 
173
  gr.Interface(fn=predict,
174
- inputs=[gr.Image(type="pil",label="Input Image"),gr.Video(label="Input Video"),gr.Textbox(placeholder="0.50",label="Set the confidence threshold (0.00-1.00)")],
175
- outputs=[gr.Image(type="pil",label="Output Image"),gr.Video(label="Output Video")],
176
  title="Facemask & Glasses",
177
  description="Model: ssd_mobilenet_v2_320x320",
178
  theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
 
171
  # predicted_img.save('predicted.jpg')
172
 
173
  gr.Interface(fn=predict,
174
+ inputs=[gr.Image(type="pil",label="Input Image",height=500,width=500),gr.Video(label="Input Video",height=500,width=500),gr.Textbox(placeholder="0.50",label="Set the confidence threshold (0.00-1.00)")],
175
+ outputs=[gr.Image(type="pil",label="Output Image",height=500,width=500),gr.Video(label="Output Video",height=500,width=500)],
176
  title="Facemask & Glasses",
177
  description="Model: ssd_mobilenet_v2_320x320",
178
  theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),