warhawkmonk commited on
Commit
cbfad5a
·
verified ·
1 Parent(s): b8b337c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -100,7 +100,7 @@ def encoding_model():
100
 
101
  @st.cache_resource
102
  def Q_and_A_model():
103
- qa_model = pipeline('question-answering', model='etalab-ia/camembert-base-squadFR-fquad-piaf', tokenizer='etalab-ia/camembert-base-squadFR-fquad-piaf')
104
  return qa_model
105
 
106
  def executer(query):
@@ -809,7 +809,8 @@ if bg_doc and prompt:
809
  accurate_score= {'score':0 ,"value":""}
810
  for select in retrieved_chunks[:3]:
811
 
812
- result_test = Q_and_A_model()(prompt, select[-1])
 
813
  if result_test['score']>accurate_score['score']:
814
  accurate_score['score'] = result_test['score']
815
  accurate_score['value'] = select[-1]
 
100
 
101
  @st.cache_resource
102
  def Q_and_A_model():
103
+ qa_model = pipeline('question-answering', model='CATIE-AQ/QAmembert-large', tokenizer='CATIE-AQ/QAmembert-large')
104
  return qa_model
105
 
106
  def executer(query):
 
809
  accurate_score= {'score':0 ,"value":""}
810
  for select in retrieved_chunks[:3]:
811
 
812
+ result_test = Q_and_A_model()
813
+ result_test=result_test(prompt, select[-1])
814
  if result_test['score']>accurate_score['score']:
815
  accurate_score['score'] = result_test['score']
816
  accurate_score['value'] = select[-1]