DevBM commited on
Commit
65b7df9
·
verified ·
1 Parent(s): 74406b7

Update app.py

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