Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -66,7 +66,7 @@ def set_state(session_id, key, value):
|
|
66 |
|
67 |
@st.cache_resource
|
68 |
def load_model():
|
69 |
-
model_name = "DevBM/t5-
|
70 |
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
71 |
tokenizer = T5Tokenizer.from_pretrained(model_name)
|
72 |
return model, tokenizer
|
|
|
66 |
|
67 |
@st.cache_resource
|
68 |
def load_model():
|
69 |
+
model_name = "DevBM/t5-large-squad"
|
70 |
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
71 |
tokenizer = T5Tokenizer.from_pretrained(model_name)
|
72 |
return model, tokenizer
|