imports

app.py

@@ -8,6 +8,7 @@ import albumentations as A
 from albumentations.pytorch import ToTensorV2
 from PIL import Image
 from transformers import ConvNextV2Config, ConvNextV2ForImageClassification
+import numpy as np
 
 # preprocessing
 transforms = A.Compose([