Update utils.py

utils.py

@@ -43,6 +43,9 @@ def build_dataset(base_path):
 
 feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = AutoModelForImageClassification.from_pretrained("microsoft/resnet-50")
+model.to(device)
+model.eval()
 
 # Fonction load_and_preprocess_image (inchangée)
 def load_and_preprocess_image(img_path):