Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
cc0725b
1
Parent(s):
4af3c79
Update train_model.py
Browse files
src/models/train_model.py
CHANGED
@@ -44,6 +44,7 @@ def train_model():
|
|
44 |
json.dump(data, fp)
|
45 |
|
46 |
if params["upload_to_hf"]:
|
|
|
47 |
|
48 |
if __name__ == "__main__":
|
49 |
train_model()
|
|
|
44 |
json.dump(data, fp)
|
45 |
|
46 |
if params["upload_to_hf"]:
|
47 |
+
model.upload(hf_username=params["hf_username"], model_name=params["name"])
|
48 |
|
49 |
if __name__ == "__main__":
|
50 |
train_model()
|