Update

.pre-commit-config.yaml

@@ -21,26 +21,17 @@ repos:
   - id: docformatter
     args: ['--in-place']
 - repo: https://github.com/pycqa/isort
-  rev: 5.10.1
+  rev: 5.12.0
   hooks:
     - id: isort
 - repo: https://github.com/pre-commit/mirrors-mypy
-  rev: v0.812
+  rev: v0.991
   hooks:
     - id: mypy
       args: ['--ignore-missing-imports']
+      additional_dependencies: ['types-python-slugify']
 - repo: https://github.com/google/yapf
   rev: v0.32.0
   hooks:
   - id: yapf
     args: ['--parallel', '--in-place']
-- repo: https://github.com/kynan/nbstripout
-  rev: 0.5.0
-  hooks:
-    - id: nbstripout
-      args: ['--extra-keys', 'metadata.interpreter metadata.kernelspec cell.metadata.pycharm']
-- repo: https://github.com/nbQA-dev/nbQA
-  rev: 1.3.1
-  hooks:
-    - id: nbqa-isort
-    - id: nbqa-yapf

README.md

@@ -4,9 +4,12 @@ emoji: 📊
 colorFrom: yellow
 colorTo: indigo
 sdk: gradio
-sdk_version: 3.21.0
+sdk_version: 3.35.2
 app_file: app.py
 pinned: false
+suggested_hardware: t4-small
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+
+https://arxiv.org/abs/2204.12484

app.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import argparse
 import pathlib
 import tarfile
 
@@ -13,10 +12,6 @@ from model import AppDetModel, AppPoseModel
 DESCRIPTION = '# [ViTPose](https://github.com/ViTAE-Transformer/ViTPose)'
 
 
-def set_example_image(example: list) -> dict:
-    return gr.Image.update(value=example[0])
-
-
 def extract_tar() -> None:
     if pathlib.Path('mmdet_configs/configs').exists():
         return
@@ -160,4 +155,4 @@ with gr.Blocks(css='style.css') as demo:
                              ],
                              outputs=pose_visualization)
 
-demo.queue(api_open=False).launch()
+demo.queue(max_size=10).launch()

model.py

@@ -14,7 +14,7 @@ if os.getenv('SYSTEM') == 'spaces':
 
     subprocess.run(shlex.split('pip uninstall -y opencv-python'))
     subprocess.run(shlex.split('pip uninstall -y opencv-python-headless'))
-    subprocess.run(shlex.split('pip install opencv-python-headless==4.5.5.64'))
+    subprocess.run(shlex.split('pip install opencv-python-headless==4.8.0.74'))
 
 import huggingface_hub
 import numpy as np
@@ -29,8 +29,6 @@ from mmdet.apis import inference_detector, init_detector
 from mmpose.apis import (inference_top_down_pose_model, init_pose_model,
                          process_mmdet_results, vis_pose_result)
 
-HF_TOKEN = os.getenv('HF_TOKEN')
-
 
 class DetModel:
     MODEL_DICT = {
@@ -72,8 +70,8 @@ class DetModel:
             self._load_model(name)
 
     def _load_model(self, name: str) -> nn.Module:
-        dic = self.MODEL_DICT[name]
-        return init_detector(dic['config'], dic['model'], device=self.device)
+        d = self.MODEL_DICT[name]
+        return init_detector(d['config'], d['model'], device=self.device)
 
     def set_model(self, name: str) -> None:
         if name == self.model_name:
@@ -152,11 +150,10 @@ class PoseModel:
             self._load_model(name)
 
     def _load_model(self, name: str) -> nn.Module:
-        dic = self.MODEL_DICT[name]
-        ckpt_path = huggingface_hub.hf_hub_download('hysts/ViTPose',
-                                                    dic['model'],
-                                                    use_auth_token=HF_TOKEN)
-        model = init_pose_model(dic['config'], ckpt_path, device=self.device)
+        d = self.MODEL_DICT[name]
+        ckpt_path = huggingface_hub.hf_hub_download('public-data/ViTPose',
+                                                    d['model'])
+        model = init_pose_model(d['config'], ckpt_path, device=self.device)
         return model
 
     def set_model(self, name: str) -> None:

requirements.txt

@@ -2,7 +2,7 @@ mmcv-full==1.5.0
 mmdet==2.24.1
 mmpose==0.25.1
 numpy==1.23.5
-opencv-python-headless==4.5.5.64
+opencv-python-headless==4.8.0.74
 openmim==0.1.5
 timm==0.5.4
 torch==1.11.0