DevBM commited on
Commit
f7dcaaa
1 Parent(s): d7d7350

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ print("***************************************************************")
30
 
31
  st.set_page_config(
32
  page_title="Question Generator",
33
- initial_sidebar_state="collapsed",
34
  )
35
 
36
  # Initialize Wikipedia API with a user agent
@@ -40,7 +40,7 @@ wiki_wiki = wikipediaapi.Wikipedia(user_agent= user_agent,language='en')
40
 
41
  @st.cache_resource
42
  def load_model():
43
- model_name = "DevBM/t5-small-squad"
44
  model = T5ForConditionalGeneration.from_pretrained(model_name)
45
  tokenizer = T5Tokenizer.from_pretrained(model_name)
46
  return model, tokenizer
 
30
 
31
  st.set_page_config(
32
  page_title="Question Generator",
33
+ initial_sidebar_state="auto",
34
  )
35
 
36
  # Initialize Wikipedia API with a user agent
 
40
 
41
  @st.cache_resource
42
  def load_model():
43
+ model_name = "DevBM/t5-large-squad"
44
  model = T5ForConditionalGeneration.from_pretrained(model_name)
45
  tokenizer = T5Tokenizer.from_pretrained(model_name)
46
  return model, tokenizer