Spaces:
Runtime error
Runtime error
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
|