alexkueck commited on
Commit
37b019f
·
1 Parent(s): b51d8fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -105,7 +105,7 @@ print ("lm datasets")
105
  batch_size = 2
106
 
107
  training_args = TrainingArguments(
108
- output_dir="C:/Users/alexk/Documents/Schule",
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("C:/Users/alexk/Documents/Schule")
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/model/finetune-tis")
183
- #print("done")
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