Spaces:
Running
Running
Update app.py
Browse files
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='
|
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()
|
|
|
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]
|