nicodioub commited on
Commit
fe6cf06
·
verified ·
1 Parent(s): c48af1e

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -0
utils.py CHANGED
@@ -41,6 +41,8 @@ def build_dataset(base_path):
41
  "category": categories
42
  })
43
 
 
 
44
  # Fonction load_and_preprocess_image (inchangée)
45
  def load_and_preprocess_image(img_path):
46
  """
 
41
  "category": categories
42
  })
43
 
44
+ feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
45
+
46
  # Fonction load_and_preprocess_image (inchangée)
47
  def load_and_preprocess_image(img_path):
48
  """