Cicciokr commited on
Commit
5ceccd9
·
verified ·
1 Parent(s): 25d41aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-CC100-la"
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