Update app.py

app.py

@@ -49,8 +49,8 @@ input_text = st.text_area(
 #tokenizer = GPT2Tokenizer.from_pretrained("Cicciokr/GPT2-Latin-GenText")
 #model_name = "morenolq/bart-it"
 model_name_mio = "Cicciokr/BART-la-s"
-model_mio = BartForConditionalGeneration.from_pretrained(model_name_mio)
-tokenizer_mio = BartTokenizer.from_pretrained(model_name_mio)
+model_mio = AutoModelForSeq2SeqLM.from_pretrained(model_name_mio)
+tokenizer_mio = AutoTokenizer.from_pretrained(model_name_mio)
 tokenizer_mio.pad_token_id = tokenizer_mio.eos_token_id
 
 #generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)