Irina Tolstykh commited on
Commit
85e6850
1 Parent(s): 86551c8

update version 2

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +3 -3
.gitignore CHANGED
@@ -1,2 +1,3 @@
1
  gradio_cached_examples/
2
  flagged/
 
 
1
  gradio_cached_examples/
2
  flagged/
3
+ .idea/
app.py CHANGED
@@ -49,7 +49,7 @@ def load_models():
49
  use_auth_token=HF_TOKEN)
50
 
51
  age_gender_path_v2 = huggingface_hub.hf_hub_download('iitolstykh/demo_xnet_volo_cross',
52
- 'mivolo_v2_1.tar',
53
  use_auth_token=HF_TOKEN)
54
 
55
  predictor_cfg_v1 = Cfg(detector_path, age_gender_path_v1)
@@ -164,8 +164,8 @@ with gr.Blocks(theme=gr.themes.Default(), css="style.css") as demo:
164
  with gr.Tabs():
165
  with gr.Tab(label="MiVOLO_V1"):
166
  demo_v1.render()
167
- # with gr.Tab(label="MiVOLO_V2"):
168
- # demo_v2.render()
169
 
170
 
171
  if __name__ == "__main__":
 
49
  use_auth_token=HF_TOKEN)
50
 
51
  age_gender_path_v2 = huggingface_hub.hf_hub_download('iitolstykh/demo_xnet_volo_cross',
52
+ 'mivolo_v2_384_0.15.pth.tar',
53
  use_auth_token=HF_TOKEN)
54
 
55
  predictor_cfg_v1 = Cfg(detector_path, age_gender_path_v1)
 
164
  with gr.Tabs():
165
  with gr.Tab(label="MiVOLO_V1"):
166
  demo_v1.render()
167
+ with gr.Tab(label="MiVOLO_V2"):
168
+ demo_v2.render()
169
 
170
 
171
  if __name__ == "__main__":