narugo1992 commited on
Commit
f4177d4
1 Parent(s): 726a752

dev(narugo): add new model of caihongmao

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -98,6 +98,7 @@ def image_to_tensor(pic: Image.Image, size: int = 512, keep_ratio: float = True,
98
 
99
 
100
  MODELS = [
 
101
  'TResnet-D-FLq_ema_6-30000.onnx',
102
  'TResnet-D-FLq_ema_6-10000.onnx',
103
  'TResnet-D-FLq_ema_4-10000.onnx',
@@ -180,7 +181,8 @@ if __name__ == '__main__':
180
  gr_input_image = gr.Image(type='pil', label='Original Image')
181
  with gr.Row():
182
  gr_threshold = gr.Slider(0.0, 1.0, 0.7, label='Tagging Confidence Threshold')
183
- gr_image_size = gr.Slider(128, 960, 640, step=32, label='Image for Recognition')
 
184
  gr_keep_ratio = gr.Checkbox(value=False, label='Keep the Ratio')
185
  with gr.Row():
186
  gr_model = gr.Dropdown(MODELS, value=DEFAULT_MODEL, label='Model')
 
98
 
99
 
100
  MODELS = [
101
+ 'caformer_m36-3-80000.onnx',
102
  'TResnet-D-FLq_ema_6-30000.onnx',
103
  'TResnet-D-FLq_ema_6-10000.onnx',
104
  'TResnet-D-FLq_ema_4-10000.onnx',
 
181
  gr_input_image = gr.Image(type='pil', label='Original Image')
182
  with gr.Row():
183
  gr_threshold = gr.Slider(0.0, 1.0, 0.7, label='Tagging Confidence Threshold')
184
+ # gr_image_size = gr.Slider(128, 960, 640, step=32, label='Image for Recognition')
185
+ gr_image_size = gr.Slider(128, 960, 448, step=32, label='Image for Recognition')
186
  gr_keep_ratio = gr.Checkbox(value=False, label='Keep the Ratio')
187
  with gr.Row():
188
  gr_model = gr.Dropdown(MODELS, value=DEFAULT_MODEL, label='Model')