Spaces:
Sleeping
Sleeping
Truong-Phuc Nguyen
commited on
Commit
•
9c408a6
1
Parent(s):
d0150b0
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ st.set_page_config(page_title="Vietnamese Legal Question Answering", page_icon="
|
|
8 |
device = 0 if torch.cuda.is_available() else -1
|
9 |
if 'model' not in st.session_state:
|
10 |
print('Some errors occured!')
|
11 |
-
st.session_state.model = pipeline("question-answering", model='
|
12 |
|
13 |
def get_answer(context, question):
|
14 |
return st.session_state.model(context=context, question=question, max_answer_len=512)
|
|
|
8 |
device = 0 if torch.cuda.is_available() else -1
|
9 |
if 'model' not in st.session_state:
|
10 |
print('Some errors occured!')
|
11 |
+
st.session_state.model = pipeline("question-answering", model='Truong-Phuc/ViL-MRC-large', device=device)
|
12 |
|
13 |
def get_answer(context, question):
|
14 |
return st.session_state.model(context=context, question=question, max_answer_len=512)
|