Spaces:
Sleeping
Sleeping
okeowo1014
commited on
Commit
•
300365a
1
Parent(s):
4fac414
Update train.py
Browse files
train.py
CHANGED
@@ -112,7 +112,7 @@ model.save('cat_dog_classifierss.keras','',save_format='tf',)
|
|
112 |
# from huggingface_hub import hf_hub_auth
|
113 |
# hf_hub_auth(token=sac)
|
114 |
# Recreate the exact same model purely from the file
|
115 |
-
new_model = keras.models.load_model('cat_dog_classifierss.keras')
|
116 |
# Push the model to Hugging Face Hub with a descriptive name
|
117 |
model_repo = push_to_hub_keras(new_model, repo_id="okeowo1014/kerascatanddog",token=sac) # Optional organization
|
118 |
|
|
|
112 |
# from huggingface_hub import hf_hub_auth
|
113 |
# hf_hub_auth(token=sac)
|
114 |
# Recreate the exact same model purely from the file
|
115 |
+
new_model = tf.keras.models.load_model('cat_dog_classifierss.keras')
|
116 |
# Push the model to Hugging Face Hub with a descriptive name
|
117 |
model_repo = push_to_hub_keras(new_model, repo_id="okeowo1014/kerascatanddog",token=sac) # Optional organization
|
118 |
|