Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,9 +10,9 @@ st.subheader("Try")
|
|
10 |
@st.cache_resource(show_spinner=True)
|
11 |
def question_model():
|
12 |
model_name = "kxx-kkk/FYP_deberta-v3-base-squad2_mrqa"
|
13 |
-
|
14 |
model = AutoModelForQuestionAnswering.from_pretrained(model_name)
|
15 |
-
question_answerer = pipeline("question-answering", model=model)
|
16 |
return question_answerer
|
17 |
|
18 |
st.markdown("<h2 style='text-align: center; color:grey;'>Question Answering on Academic Essays</h2>", unsafe_allow_html=True)
|
|
|
10 |
@st.cache_resource(show_spinner=True)
|
11 |
def question_model():
|
12 |
model_name = "kxx-kkk/FYP_deberta-v3-base-squad2_mrqa"
|
13 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
14 |
model = AutoModelForQuestionAnswering.from_pretrained(model_name)
|
15 |
+
question_answerer = pipeline("question-answering", model=model, tokenizer=tokenizer)
|
16 |
return question_answerer
|
17 |
|
18 |
st.markdown("<h2 style='text-align: center; color:grey;'>Question Answering on Academic Essays</h2>", unsafe_allow_html=True)
|