orhir / PoseAnything

A Graph-Based Approach for Category-Agnostic Pose Estimation [ECCV 2024]
https://orhir.github.io/pose-anything/
Apache License 2.0
308 stars 17 forks source link

problems when running the code in colab #5

Closed AmitIsraeli closed 9 months ago

AmitIsraeli commented 9 months ago

Hi guys love your work (: I am trying to run your code(the gradio demo) on colab and I am getting a problem. I would love if you guys could upload a official colab or a huggingface space. File "/usr/local/lib/python3.10/dist-packages/gradio/routes.py", line 508, in predict output = await route_utils.call_process_api( File "/usr/local/lib/python3.10/dist-packages/gradio/route_utils.py", line 219, in call_process_api output = await app.get_blocks().process_api( File "/usr/local/lib/python3.10/dist-packages/gradio/blocks.py", line 1437, in process_api result = await self.call_function( File "/usr/local/lib/python3.10/dist-packages/gradio/blocks.py", line 1109, in call_function prediction = await anyio.to_thread.run_sync( File "/usr/local/lib/python3.10/dist-packages/anyio/to_thread.py", line 33, in run_sync return await get_asynclib().run_sync_in_worker_thread( File "/usr/local/lib/python3.10/dist-packages/anyio/_backends/_asyncio.py", line 877, in run_sync_in_worker_thread return await future File "/usr/local/lib/python3.10/dist-packages/anyio/_backends/_asyncio.py", line 807, in run result = context.run(func, *args) File "/usr/local/lib/python3.10/dist-packages/gradio/utils.py", line 641, in wrapper response = f(*args, **kwargs) File "/content/PoseAnything/app.py", line 153, in process load_checkpoint(model, checkpoint_path, map_location='cpu') File "/usr/local/lib/python3.10/dist-packages/mmcv/runner/checkpoint.py", line 638, in load_checkpoint checkpoint = _load_checkpoint(filename, map_location, logger) File "/usr/local/lib/python3.10/dist-packages/mmcv/runner/checkpoint.py", line 572, in _load_checkpoint return CheckpointLoader.load_checkpoint(filename, map_location, logger) File "/usr/local/lib/python3.10/dist-packages/mmcv/runner/checkpoint.py", line 314, in load_checkpoint return checkpoint_loader(filename, map_location) # type: ignore File "/usr/local/lib/python3.10/dist-packages/mmcv/runner/checkpoint.py", line 334, in load_from_local checkpoint = torch.load(filename, map_location=map_location) File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 993, in load with _open_zipfile_reader(opened_file) as opened_zipfile: File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 447, in __init__ super().__init__(torch._C.PyTorchFileReader(name_or_buffer))

orhir commented 9 months ago

Thank you for showing interest in our work. We are currently working on integrating our demo to huggingface.

In the meantime, we have uploaded the demo into OpenXLab, which is available for online use: https://openxlab.org.cn/apps/detail/orhir/Pose-Anything

orhir commented 9 months ago

We are now also available on Huggingface: https://huggingface.co/spaces/orhir/PoseAnything