Spaces:

okeowo1014
/

catsandogs

Sleeping

okeowo1014 commited on Mar 20

Commit

b65427f

•

1 Parent(s): 300365a

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -106,7 +106,7 @@ model.summary()
 #     token=sac,
 # )
 # Save the model with a valid extension (recommended: .h5)
-model.save('cat_dog_classifierss.keras','',save_format='tf',)
 # tf.keras.models.save_model(model, 'cat_dog_classifierss.keras')
 # Authenticate with Hugging Face (replace YOUR_TOKEN with your actual token)
 # from huggingface_hub import hf_hub_auth

 #     token=sac,
 # )
 # Save the model with a valid extension (recommended: .h5)
+model.save('cat_dog_classifierss.keras')
 # tf.keras.models.save_model(model, 'cat_dog_classifierss.keras')
 # Authenticate with Hugging Face (replace YOUR_TOKEN with your actual token)
 # from huggingface_hub import hf_hub_auth