Spaces:

OOlajide
/

common-nlp-tasks

Sleeping

OOlajide commited on Dec 29, 2021

Commit

1ae1e57

1 Parent(s): 5ac3f90

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ option = st.sidebar.radio('', ['Extractive question answering', 'Text summarizat
 @st.cache(show_spinner=False, allow_output_mutation=True)
 def question_model():
     tokenizer = AutoTokenizer.from_pretrained("deepset/roberta-base-squad2")
-    model = AutoModelForQuestionAnswering.from_pretrained("deepset/roberta-base-squad2")
     return question_answerer
 @st.cache(show_spinner=False, allow_output_mutation=True)

 @st.cache(show_spinner=False, allow_output_mutation=True)
 def question_model():
     tokenizer = AutoTokenizer.from_pretrained("deepset/roberta-base-squad2")
+    question_answerer = AutoModelForQuestionAnswering.from_pretrained("deepset/roberta-base-squad2")
     return question_answerer
 @st.cache(show_spinner=False, allow_output_mutation=True)