Irina Tolstykh
commited on
Commit
•
85e6850
1
Parent(s):
86551c8
update version 2
Browse files- .gitignore +1 -0
- 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 |
-
'
|
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 |
-
|
168 |
-
|
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__":
|