pjdevelop
/

dinov2-svm-embeddings-currency_indian

Image Classification

Inference Endpoints

Model card Files Files and versions Community

pjdevelop commited on Jul 28, 2024

Commit

629fcd8

·

verified ·

1 Parent(s): 00ce9dd

Update model.py

Files changed (1) hide show

model.py +2 -4

model.py CHANGED Viewed

@@ -3,14 +3,12 @@ import torchvision.transforms as T
 from PIL import Image
 import joblib
 import json
-import cv2
-import gradio as gr
 # Define image transformation
 transform_image = T.Compose([
-    T.ToTensor(),
-    T.Resize(244),
     T.CenterCrop(224),
     T.Normalize([0.5], [0.5])
 ])

 from PIL import Image
 import joblib
 import json
 # Define image transformation
 transform_image = T.Compose([
+    T.Resize(224),  # Corrected to 224 to match CenterCrop
     T.CenterCrop(224),
+    T.ToTensor(),
     T.Normalize([0.5], [0.5])
 ])