search

niki-amini-naieni / CountGD

Includes the code for training and testing the CountGD model from the paper CountGD: Multi-Modal Open-World Counting.
MIT License
90 stars 8 forks source link

One-line code api for model #2

Closed Dimlife closed 3 months ago

Dimlife commented 3 months ago

Hi, thanks for sharing the amazing work! I am wondering could we have an one-line-command build or a simple api of the model? Really appreciate that.

niki-amini-naieni commented 3 months ago

Hi @Dimlife, thank you for the comment! Have you seen the Hugging Face app that includes simple code in the file app.py as well as a user friendly tutorial and interface to test the model? It is available here: https://huggingface.co/spaces/nikigoli/countgd. You can run the app with the docker command below from hugging face.

docker run -it -p 7860:7860 --platform=linux/amd64 --gpus all \
    registry.hf.space/nikigoli-countgd:latest python app.py
Dimlife commented 3 months ago

Got this. Thank you very much for replying!