pere commited on
Commit
690e0ae
·
1 Parent(s): 90a6c0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import T5ForConditionalGeneration, T5TokenizerFast, T5Config
4
  @st.cache(allow_output_mutation=True, suppress_st_warning=True)
5
  def load_model():
6
  model_name = "north/demo-nynorsk-base"
7
- config= T5Config.from_pretrained(model_name+'/config.json')
8
  model = T5ForConditionalGeneration.from_pretrained(model_name,config=config)
9
  tokenizer = T5TokenizerFast.from_pretrained(model_name)
10
  return (model, tokenizer)
 
4
  @st.cache(allow_output_mutation=True, suppress_st_warning=True)
5
  def load_model():
6
  model_name = "north/demo-nynorsk-base"
7
+ config = T5Config.from_pretrained(model_name)
8
  model = T5ForConditionalGeneration.from_pretrained(model_name,config=config)
9
  tokenizer = T5TokenizerFast.from_pretrained(model_name)
10
  return (model, tokenizer)