Fixes airctic#1160 Allow SAHI models to load for colab / intro tutorials

n-e-w · Dec 18, 2022 · cefb09e · cefb09e
1 parent ffc0815
commit cefb09e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/icevision/models/inference_sahi.py b/icevision/models/inference_sahi.py
@@ -11,7 +11,7 @@
 from icevision.tfms.albumentations import albumentations_adapter
 
 
-from sahi.models.base import DetectionModel
+from sahi import DetectionModel
 from sahi.prediction import ObjectPrediction
 from sahi.predict import get_sliced_prediction as sahi_get_sliced_prediction