Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ print("***************************************************************")
|
|
30 |
|
31 |
st.set_page_config(
|
32 |
page_title="Question Generator",
|
33 |
-
initial_sidebar_state="
|
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-
|
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
|