alexkueck commited on
Commit
3ea4315
·
1 Parent(s): 60d30cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ lm_datasets = tokenized_datasets.map(
88
  #Training Args
89
  model_name = base_model.split("/")[-1]
90
  training_args = TrainingArguments(
91
- f"{model_name}-finetuned-tis",
92
  evaluation_strategy = "epoch",
93
  learning_rate=2e-5,
94
  weight_decay=0.01,
 
88
  #Training Args
89
  model_name = base_model.split("/")[-1]
90
  training_args = TrainingArguments(
91
+ "alexkueck/tis-trained-model",
92
  evaluation_strategy = "epoch",
93
  learning_rate=2e-5,
94
  weight_decay=0.01,