Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -105,7 +105,7 @@ print ("lm datasets")
|
|
105 |
batch_size = 2
|
106 |
|
107 |
training_args = TrainingArguments(
|
108 |
-
output_dir="
|
109 |
overwrite_output_dir = 'True',
|
110 |
per_device_train_batch_size=batch_size, #batch_size = 2 for full training
|
111 |
per_device_eval_batch_size=batch_size,
|
@@ -172,15 +172,15 @@ print("Output:\n" )
|
|
172 |
|
173 |
|
174 |
print("Save to Space")
|
175 |
-
trainer.save_model("
|
176 |
print("done")
|
177 |
|
178 |
#####################################
|
179 |
#Push to Hub
|
180 |
-
|
181 |
-
|
182 |
-
|
183 |
-
|
184 |
|
185 |
|
186 |
|
|
|
105 |
batch_size = 2
|
106 |
|
107 |
training_args = TrainingArguments(
|
108 |
+
output_dir="model",
|
109 |
overwrite_output_dir = 'True',
|
110 |
per_device_train_batch_size=batch_size, #batch_size = 2 for full training
|
111 |
per_device_eval_batch_size=batch_size,
|
|
|
172 |
|
173 |
|
174 |
print("Save to Space")
|
175 |
+
trainer.save_model("model")
|
176 |
print("done")
|
177 |
|
178 |
#####################################
|
179 |
#Push to Hub
|
180 |
+
print("push to hub")
|
181 |
+
login(token=os.environ["HF_ACCESS_TOKEN"])
|
182 |
+
trainer.push_to_hub("alexkueck/test-tis")
|
183 |
+
print("done")
|
184 |
|
185 |
|
186 |
|