shvardhan commited on
Commit
8353a27
·
1 Parent(s): 8dda508

update app file

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -30,9 +30,7 @@ This is an unofficial demo for [https://github.com/open-mmlab/mmdetection](https
30
 
31
  DEFAULT_MODEL_TYPE = 'detection'
32
  DEFAULT_MODEL_NAMES = {
33
- 'detection': 'YOLOX-l',
34
- 'instance_segmentation': 'QueryInst (R-50-FPN)',
35
- 'panoptic_segmentation': 'MaskFormer (R-50)',
36
  }
37
  DEFAULT_MODEL_NAME = DEFAULT_MODEL_NAMES[DEFAULT_MODEL_TYPE]
38
 
@@ -82,7 +80,7 @@ with gr.Blocks(css='style.css') as demo:
82
  label='Model Type')
83
  with gr.Row():
84
  model_name = gr.Dropdown(list(
85
- model.DETECTION_MODEL_DICT.keys()),
86
  value=DEFAULT_MODEL_NAME,
87
  label='Model')
88
  with gr.Row():
 
30
 
31
  DEFAULT_MODEL_TYPE = 'detection'
32
  DEFAULT_MODEL_NAMES = {
33
+ 'detection': 'Faster-RCNN',
 
 
34
  }
35
  DEFAULT_MODEL_NAME = DEFAULT_MODEL_NAMES[DEFAULT_MODEL_TYPE]
36
 
 
80
  label='Model Type')
81
  with gr.Row():
82
  model_name = gr.Dropdown(list(
83
+ 'Faster R-CNN (R-50-FPN)'),
84
  value=DEFAULT_MODEL_NAME,
85
  label='Model')
86
  with gr.Row():