Installation issue

[anilsathyan7]

Tried demo code in google colab, but it seems to fail as follows:

!pip install autodistill autodistill-grounded-sam autodistill-yolov8
!pip3 install autodistill-sam-hq roboflow supervision

from autodistill_sam_hq import HQSAM
base_model = HQSAM(None)
masks = base_model.predict("eb4c9db3a9b8e3bf6cb8397bc00ad075.png")
print(masks)

ModuleNotFoundError                       Traceback (most recent call last)
[<ipython-input-2-3512cfbd3532>](https://localhost:8080/#) in <cell line: 4>()
      2 get_ipython().system('pip3 install autodistill-sam-hq roboflow supervision')
      3 
----> 4 from autodistill_sam_hq import HQSAM
      5 
      6 base_model = HQSAM(None)

1 frames
[/usr/local/lib/python3.10/dist-packages/autodistill_sam_hq/samhq_model.py](https://localhost:8080/#) in <module>
      7 import supervision as sv
      8 import torch
----> 9 from segment_anything_hq import SamAutomaticMaskGenerator, sam_model_registry
     10 
     11 from autodistill.detection import CaptionOntology, DetectionBaseModel

ModuleNotFoundError: No module named 'segment_anything_hq'