Hugging Face's logo

Spaces:
devdata
/
kapu
Runtime error

App Files Community
devdata commited on
Commit
df7a0c9
·
1 Parent(s): e087b23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from transformers import MobileNetV2FeatureExtractor, MobileNetV2ForImageClassif
2
  from PIL import Image
3
 
4
  # Replace with the path to the directory containing the model files
5
- model_folder = "/path/to/your/model_folder"
6
 
7
  # Load the feature extractor
8
  feature_extractor = MobileNetV2FeatureExtractor.from_pretrained(model_folder)
 
2
  from PIL import Image
3
 
4
  # Replace with the path to the directory containing the model files
5
+ model_folder = "mobilnetV2_ftprecmodelshuf.weights.best.hdf5"
6
 
7
  # Load the feature extractor
8
  feature_extractor = MobileNetV2FeatureExtractor.from_pretrained(model_folder)