Update app.py
Browse files
app.py
CHANGED
|
@@ -49,7 +49,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
|
|
| 49 |
batch_size = 8
|
| 50 |
|
| 51 |
args = Seq2SeqTrainingArguments(
|
| 52 |
-
f"{
|
| 53 |
evaluation_strategy="epoch",
|
| 54 |
learning_rate=2e-5,
|
| 55 |
per_device_train_batch_size=batch_size,
|
|
|
|
| 49 |
batch_size = 8
|
| 50 |
|
| 51 |
args = Seq2SeqTrainingArguments(
|
| 52 |
+
f"{model_checkpoint}-scientific_papers",
|
| 53 |
evaluation_strategy="epoch",
|
| 54 |
learning_rate=2e-5,
|
| 55 |
per_device_train_batch_size=batch_size,
|