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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -162,7 +162,7 @@ def video_fn(video_in_filepath):
162
  cv2.waitKey(1)
163
  return video_out_filepath
164
 
165
- REPO_ID = "23b719w/assignment2_tfodmodel"
166
  detection_model = load_model()
167
  # pil_image = Image.open(image_path)
168
  # image_arr = pil_image_as_numpy_array(pil_image)
@@ -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",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"),
 
162
  cv2.waitKey(1)
163
  return video_out_filepath
164
 
165
+ REPO_ID = "23b719w/assignment2tfod_model2"
166
  detection_model = load_model()
167
  # pil_image = Image.open(image_path)
168
  # image_arr = pil_image_as_numpy_array(pil_image)
 
171
  # predicted_img.save('predicted.jpg')
172
 
173
  gr.Interface(fn=predict,
174
+ inputs=[gr.Image(type="pil",label="Input Image",height=500,width=800),gr.Video(label="Input Video",height=500,width=800),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=800),gr.Video(label="Output Video",height=500,width=800)],
176
  title="Facemask & Glasses",
177
  description="Model: ssd_mobilenet_v2_320x320",
178
  theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),