Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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.
|
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
|