Spaces:

josueadin
/

dog_breed_classifier

Runtime error

adinmg commited on Sep 12, 2023

Commit

07835b2

1 Parent(s): d462f3f

model path modified

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ with open('class_names.txt') as f:
     class_names = [breed.strip() for breed in f.readlines()]
 # Load the pre-trained EfficientNetV2 model
-effnet = tensorflow.keras.models.load_model('demo/dog_breed_classifier/dog_breed_effnet_augmentation.h5')
 # Get the preprocessing function for EfficientNetV2
 effnet_preprocess_input = tensorflow.keras.applications.efficientnet_v2.preprocess_input

     class_names = [breed.strip() for breed in f.readlines()]
 # Load the pre-trained EfficientNetV2 model
+effnet = tensorflow.keras.models.load_model('dog_breed_effnet_augmentation.h5')
 # Get the preprocessing function for EfficientNetV2
 effnet_preprocess_input = tensorflow.keras.applications.efficientnet_v2.preprocess_input