shvardhan commited on
Commit
3295878
1 Parent(s): 6547459

update app file

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -42,6 +42,7 @@ def update_input_image(image: np.ndarray) -> dict:
42
  scale = 1500 / max(image.shape[:2])
43
  if scale < 1:
44
  image = cv2.resize(image, None, fx=scale, fy=scale)
 
45
  return gr.Image.update(value=image)
46
 
47
 
@@ -79,8 +80,8 @@ with gr.Blocks(css='style.css') as demo:
79
  value=DEFAULT_MODEL_TYPE,
80
  label='Model Type')
81
  with gr.Row():
82
- model_name = gr.Dropdown((
83
- 'Faster R-CNN (R-50-FPN)'),
84
  value=DEFAULT_MODEL_NAME,
85
  label='Model')
86
  with gr.Row():
 
42
  scale = 1500 / max(image.shape[:2])
43
  if scale < 1:
44
  image = cv2.resize(image, None, fx=scale, fy=scale)
45
+ print('Image shape', image.shape)
46
  return gr.Image.update(value=image)
47
 
48
 
 
80
  value=DEFAULT_MODEL_TYPE,
81
  label='Model Type')
82
  with gr.Row():
83
+ model_name = gr.Dropdown(([
84
+ 'Faster R-CNN (R-50-FPN']),
85
  value=DEFAULT_MODEL_NAME,
86
  label='Model')
87
  with gr.Row():