Update app.py

app.py

@@ -43,8 +43,8 @@ for split in ["train", "validation", "test"]:
     raw_dataset[split] = raw_dataset[split].select([n for n in np.random.randint(0, len(raw_dataset[split]) - 1, 1_000)])
 tokenized_dataset = raw_dataset.map(preprocess_function, batched=True)
 
-model_name = "fine-tuned-t5-small"
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+
+model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 
 batch_size = 8