Spaces:

dmibor
/

ietm_search_and_qa

Sleeping

dmibor commited on Oct 30, 2024

Commit

ec2e4f6

1 Parent(s): 216e1e1

bug fix

Files changed (2) hide show

__pycache__/search_core.cpython-312.pyc CHANGED Viewed

Binary files a/__pycache__/search_core.cpython-312.pyc and b/__pycache__/search_core.cpython-312.pyc differ

search_core.py CHANGED Viewed

@@ -593,9 +593,10 @@ def find_answer(question, context, threshold, max_answer_len=1000, top_k=20, ver
     #print('longest_result', longest_result)
     if best_result['score']>=threshold:
         longest_answer= longest_result['answer']
-        answer_cleaned= re.sub(r"[\W\d_]+$", "", longest_answer).strip()
         if verbose==True:
-            print(f'Ответ (уверенность= {round(model_result['score'], 2)}): {answer_cleaned}')
         longest_result['answer']= answer_cleaned
         found_answer=True
     if found_answer==False and verbose==True:

     #print('longest_result', longest_result)
     if best_result['score']>=threshold:
         longest_answer= longest_result['answer']
+        answer_cleaned= re.sub(r"[\W\d_]+$", '', longest_answer).strip()
         if verbose==True:
+            prob_value= round(model_result['score'], 2)
+            print(f'Ответ (уверенность= {prob_value}): {answer_cleaned}')
         longest_result['answer']= answer_cleaned
         found_answer=True
     if found_answer==False and verbose==True: