alexkueck commited on
Commit
35bca68
·
1 Parent(s): 5dfb843

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -104,7 +104,7 @@ print ("lm datasets")
104
  #Training Args
105
 
106
  training_args = TrainingArguments(
107
- output_dir="alexkueck/spaces/LIFineTuned/",
108
  overwrite_output_dir = 'True',
109
  evaluation_strategy = "epoch",
110
  learning_rate=2e-5,
@@ -159,7 +159,7 @@ print("Output:\n" )
159
 
160
 
161
  print("Save to Space")
162
- trainer.save_model("alexkueck/model/tis-finetuned")
163
  print("done")
164
 
165
  #####################################
 
104
  #Training Args
105
 
106
  training_args = TrainingArguments(
107
+ output_dir="alexkueck/test-tis/",
108
  overwrite_output_dir = 'True',
109
  evaluation_strategy = "epoch",
110
  learning_rate=2e-5,
 
159
 
160
 
161
  print("Save to Space")
162
+ trainer.save_model("alexkueck/test-tis")
163
  print("done")
164
 
165
  #####################################