Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
cee2223
1
Parent(s):
cc0725b
Update train_model.py
Browse files
src/models/train_model.py
CHANGED
|
@@ -46,5 +46,6 @@ def train_model():
|
|
| 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()
|
|
|
|
| 46 |
if params["upload_to_hf"]:
|
| 47 |
model.upload(hf_username=params["hf_username"], model_name=params["name"])
|
| 48 |
|
| 49 |
+
|
| 50 |
if __name__ == "__main__":
|
| 51 |
train_model()
|