fischjos commited on
Commit
00ce45e
1 Parent(s): 7291ec1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -96,7 +96,7 @@ except Exception as e:
96
  break # Debugging: check the first batch and stop
97
 
98
  # Save the model
99
- model.save('pokemon_classifier_model.h5')
100
 
101
  # Unfreeze some layers of the base model for fine-tuning
102
  base_model.trainable = True
 
96
  break # Debugging: check the first batch and stop
97
 
98
  # Save the model
99
+ model.save('pokemon_classifier_model.keras')
100
 
101
  # Unfreeze some layers of the base model for fine-tuning
102
  base_model.trainable = True