Update app.py

app.py

@@ -1,10 +1,11 @@
 import streamlit as st
-from transformers import T5ForConditionalGeneration, T5TokenizerFast
+from transformers import T5ForConditionalGeneration, T5TokenizerFast, T5Config
 
 @st.cache(allow_output_mutation=True, suppress_st_warning=True)
 def load_model():
     model_name = "north/demo-nynorsk-base"
-    model = T5ForConditionalGeneration.from_pretrained(model_name)
+    config= T5Config.from_pretrained(model_name+'/config.json')
+    model = T5ForConditionalGeneration.from_pretrained(model_name,config=config)
     tokenizer = T5TokenizerFast.from_pretrained(model_name)
     return (model, tokenizer)