Truong-Phuc Nguyen commited on
Commit
9c408a6
1 Parent(s): d0150b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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='./models/vi-mrc-large/archive/model', device=device)
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)