pytorch detector failing - No module named 'ultralytics'

[rgogada]

ERROR - Could not create detector pytorch/pytorch: No module named 'ultralytics'

here is the full log:

2023-05-18 12:44:17,621 - doods.doods - INFO - Registered detector type:tflite name:default
2023-05-18 12:44:18,382 - doods.doods - INFO - Registered detector type:tensorflow name:tensorflow
/usr/local/lib/python3.8/dist-packages/torch/hub.py:267: UserWarning: You are about to download and run code from an untrusted repository. In a future release, this won't be allowed. To add the repository to your trusted list, change the command to {calling_fn}(..., trust_repo=False) and a command prompt will appear asking for an explicit confirmation of trust, or load(..., trust_repo=True), which will assume that the prompt is to be answered with 'yes'. You can also use load(..., trust_repo='check') which will only prompt for confirmation if the repo is not already trusted. This will eventually be the default behaviour
warnings.warn(
Downloading: "https://github.com/ultralytics/yolov5/zipball/master" to /root/.cache/torch/hub/master.zip
2023-05-18 12:44:20,221 - doods.doods - ERROR - Could not create detector pytorch/pytorch: No module named 'ultralytics'
2023-05-18 12:44:20,240 - uvicorn.error - INFO - Started server process [1]
2023-05-18 12:44:20,241 - uvicorn.error - INFO - Waiting for application startup.
2023-05-18 12:44:20,241 - uvicorn.error - INFO - Application startup complete.
2023-05-18 12:44:20,241 - uvicorn.error - INFO - Uvicorn running on http://0.0.0.0:8080 (Press CTRL+C to quit)

[rgogada]

issue seems to be upstream issue
looks like there is already a PR in the queue for the fix.

mean while we can run the below inside the container to fix this issue manually.
pip install the ultralytics

restart the container

[rgogada]

here is the PR linked to that issue. ultralytics/yolov5#11529

…

On Thu, 18 May 2023 at 20:29, dtronmans ***@***.***> wrote: Hey @rgogada <https://github.com/rgogada>, Could you please guide me to the PR in the queue that you just mentioned? Best regards — Reply to this email directly, view it on GitHub <#85 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ABAWDYCZS7APEN7RS5W6RU3XGX6KBANCNFSM6AAAAAAYF5HUNA> . You are receiving this because you were mentioned.Message ID: ***@***.***>

[snowzach]

Thanks for the PR. I missed it but made the same update. Give it a shot!