Spaces:
Sleeping
Sleeping
okeowo1014
commited on
Commit
•
789498c
1
Parent(s):
9e50616
Update train.py
Browse files
train.py
CHANGED
@@ -113,6 +113,6 @@ model.save('cat_dog_classifierss.h5')
|
|
113 |
# hf_hub_auth(token=sac)
|
114 |
|
115 |
# Push the model to Hugging Face Hub with a descriptive name
|
116 |
-
model_repo = push_to_hub_keras(model,
|
117 |
|
118 |
print(f"Model pushed to Hugging Face Hub: {model_repo}")
|
|
|
113 |
# hf_hub_auth(token=sac)
|
114 |
|
115 |
# Push the model to Hugging Face Hub with a descriptive name
|
116 |
+
model_repo = push_to_hub_keras(model, repo_id="okeowo1014/kerascatanddog",token=sac) # Optional organization
|
117 |
|
118 |
print(f"Model pushed to Hugging Face Hub: {model_repo}")
|