add spaces wrappers

app.py

@@ -6,6 +6,7 @@ import threading
 import numpy as np
 import mmcv
 from time import sleep
+import spaces
 
 from mmpose.apis import inference_topdown
 from mmpose.apis import init_model as init_pose_estimator
@@ -124,6 +125,7 @@ class AsyncFrameProcessor:
                 with self.output_lock:
                     self.latest_output_frame = processed_frame
     
+    @spaces.GPU
     def _process_frame(self, frame, bbox_thr=0.3, nms_thr=0.8, kpt_thr=0.3):
         # predict bbox
         processing_start = time.time()
@@ -217,6 +219,7 @@ class AsyncFrameProcessor:
         if self.processing_thread and self.processing_thread.is_alive():
             self.processing_thread.join(timeout=2.0)
 
+    @spaces.GPU
     def init_models(self):
         # Init detector
         if self.det_model is None:
@@ -258,14 +261,14 @@ with gr.Blocks(css=css) as demo:
     gr.HTML(
         """
     <h1 style='text-align: center'>
-    Async Frame Processing Demo (Powered by WebRTC ⚡️)
+    ProbPose Webcam Demo (CVPR 2025)
     </h1>
     """
     )
     gr.HTML(
         """
         <h3 style='text-align: center'>
-        Real-time frame processing with single-slot queues
+        See <a href="https://MiraPurkrabek.github.io/ProbPose/" target="_blank">https://MiraPurkrabek.github.io/ProbPose/</a> for details.
         </h3>
         """
     )

requirements.txt

@@ -1,10 +1,10 @@
 
-## MMCV
---no-binary=mmcv,mmengine,mmdet,mmpretrain
-mmcv @ git+https://github.com/open-mmlab/[email protected]
-mmengine @ git+https://github.com/open-mmlab/[email protected]
-mmdet @ git+https://github.com/open-mmlab/[email protected]
-mmpretrain @ git+https://github.com/open-mmlab/[email protected]
+# ## MMCV
+# --no-binary=mmcv-full,mmengine,mmdet,mmpretrain
+# mmcv @ git+https://github.com/open-mmlab/[email protected]
+# mmengine @ git+https://github.com/open-mmlab/[email protected]
+# mmdet @ git+https://github.com/open-mmlab/[email protected]
+# mmpretrain @ git+https://github.com/open-mmlab/[email protected]
 
 ## build.txt
 # These must be installed before building mmpose
@@ -24,7 +24,7 @@ pillow
 scipy
 torchvision
 
---no-binary=xtcocotools
+# --no-binary=xtcocotools
 xtcocotools
 
 ## tests.txt
@@ -44,4 +44,4 @@ requests
 ## gradio app
 fastrtc
 
-mmpose @ git+https://github.com/MiraPurkrabek/ProbPose_code.git
+# mmpose @ git+https://github.com/MiraPurkrabek/ProbPose_code.git