Miroslav Purkrabek commited on
Commit
ebb18ad
·
1 Parent(s): 472855f

init models early

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -14,6 +14,7 @@ import spaces
14
  from twilio.rest import Client
15
 
16
  try:
 
17
  from mmpose.apis import inference_topdown
18
  from mmpose.apis import init_model as init_pose_estimator
19
  from mmpose.evaluation.functional import nms
@@ -24,6 +25,8 @@ except (ImportError, ModuleNotFoundError):
24
  os.system("pip uninstall -y mmpose mmdet mmcv mmengine mmpretrain")
25
  os.system("pip install -U openmim")
26
  os.system("mim install mmengine mmcv==2.1.0 mmdet==3.3.0 mmpretrain==1.2.0")
 
 
27
  from mmpose.apis import inference_topdown
28
  from mmpose.apis import init_model as init_pose_estimator
29
  from mmpose.evaluation.functional import nms
@@ -176,7 +179,7 @@ class AsyncFrameProcessor:
176
  self.visualizer = None
177
 
178
  # Initialize models using the standalone function
179
- # self.det_model, self.pose_model, self.visualizer = init_models()
180
 
181
  # Start background processing
182
  self._start_processing_thread()
 
14
  from twilio.rest import Client
15
 
16
  try:
17
+ import mmcv
18
  from mmpose.apis import inference_topdown
19
  from mmpose.apis import init_model as init_pose_estimator
20
  from mmpose.evaluation.functional import nms
 
25
  os.system("pip uninstall -y mmpose mmdet mmcv mmengine mmpretrain")
26
  os.system("pip install -U openmim")
27
  os.system("mim install mmengine mmcv==2.1.0 mmdet==3.3.0 mmpretrain==1.2.0")
28
+
29
+ import mmcv
30
  from mmpose.apis import inference_topdown
31
  from mmpose.apis import init_model as init_pose_estimator
32
  from mmpose.evaluation.functional import nms
 
179
  self.visualizer = None
180
 
181
  # Initialize models using the standalone function
182
+ self.det_model, self.pose_model, self.visualizer = init_models()
183
 
184
  # Start background processing
185
  self._start_processing_thread()