ALYYAN commited on
Commit
f017803
·
verified ·
1 Parent(s): 78cd503

Update src/cnnClassifier/pipeline/prediction.py

Browse files
src/cnnClassifier/pipeline/prediction.py CHANGED
@@ -3,6 +3,7 @@ import numpy as np
3
  from PIL import Image
4
  from transformers import AutoImageProcessor
5
  import cv2
 
6
  from mtcnn import MTCNN # For high-quality
7
  from pathlib import Path
8
  import sys
 
3
  from PIL import Image
4
  from transformers import AutoImageProcessor
5
  import cv2
6
+ from huggingface_hub import hf_hub_download
7
  from mtcnn import MTCNN # For high-quality
8
  from pathlib import Path
9
  import sys