Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,9 @@ from transformers import T5ForConditionalGeneration, T5TokenizerFast, T5Config
|
|
5 |
def load_model():
|
6 |
model_name = "north/demo-nynorsk-base"
|
7 |
config = T5Config.from_pretrained(model_name)
|
8 |
-
|
9 |
-
st.text(
|
|
|
10 |
model = T5ForConditionalGeneration.from_pretrained(model_name,config=config)
|
11 |
tokenizer = T5TokenizerFast.from_pretrained(model_name)
|
12 |
return (model, tokenizer)
|
|
|
5 |
def load_model():
|
6 |
model_name = "north/demo-nynorsk-base"
|
7 |
config = T5Config.from_pretrained(model_name)
|
8 |
+
#Debug
|
9 |
+
#st.text(config)
|
10 |
+
#st.text("north/demo-nynorsk-base")
|
11 |
model = T5ForConditionalGeneration.from_pretrained(model_name,config=config)
|
12 |
tokenizer = T5TokenizerFast.from_pretrained(model_name)
|
13 |
return (model, tokenizer)
|