Update app.py

app.py

@@ -1,11 +1,14 @@
-from transformers import MobileNetV2FeatureExtractor, MobileNetV2ForImageClassification
+from transformers import MobileNetV2FeatureExtractor, MobileNetV2ForImageClassification, MobileNetV2Config
 from PIL import Image
 
 # Replace with the path to the directory containing the model files
 model_folder = "mobilnetV2_ftprecmodelshuf.weights.best.hdf5"
 
-# Load the feature extractor
-feature_extractor = MobileNetV2FeatureExtractor.from_pretrained(model_folder)
+# Load the configuration
+config = MobileNetV2Config.from_pretrained(model_folder)
+
+# Create the feature extractor using the loaded configuration
+feature_extractor = MobileNetV2FeatureExtractor(config)
 
 # Load the model
 model = MobileNetV2ForImageClassification.from_pretrained(model_folder)