Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ input_text = st.text_area(
|
|
49 |
#model = GPT2LMHeadModel.from_pretrained("Cicciokr/GPT2-Latin-GenText")
|
50 |
#tokenizer = GPT2Tokenizer.from_pretrained("Cicciokr/GPT2-Latin-GenText")
|
51 |
#model_name = "morenolq/bart-it"
|
52 |
-
model_name_mio = "Cicciokr/BART-
|
53 |
model_mio = BartForConditionalGeneration.from_pretrained(model_name_mio)
|
54 |
tokenizer_mio = BartTokenizer.from_pretrained(model_name_mio)
|
55 |
tokenizer_mio.pad_token_id = tokenizer_mio.eos_token_id
|
|
|
49 |
#model = GPT2LMHeadModel.from_pretrained("Cicciokr/GPT2-Latin-GenText")
|
50 |
#tokenizer = GPT2Tokenizer.from_pretrained("Cicciokr/GPT2-Latin-GenText")
|
51 |
#model_name = "morenolq/bart-it"
|
52 |
+
model_name_mio = "Cicciokr/BART-la-s"
|
53 |
model_mio = BartForConditionalGeneration.from_pretrained(model_name_mio)
|
54 |
tokenizer_mio = BartTokenizer.from_pretrained(model_name_mio)
|
55 |
tokenizer_mio.pad_token_id = tokenizer_mio.eos_token_id
|