Update app.py

app.py

@@ -20,9 +20,9 @@ from langchain_core.runnables import RunnablePassthrough, RunnableParallel
 
 
 
-model_name= 'mistralai/Mistral-7B-Instruct-v0.2'
+#model_name= 'mistralai/Mistral-7B-Instruct-v0.2'
 
-#model_name='SherlockAssistant/Mistral-7B-Instruct-Ukrainian'
+model_name='SherlockAssistant/Mistral-7B-Instruct-Ukrainian'
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 tokenizer.pad_token = tokenizer.unk_token
@@ -173,7 +173,7 @@ def generate_with_filters(message, subject_input, rubric, date_beg, date_end):
     context = ret.get_relevant_documents(message)
 
     result = rag_chain_with_filters.invoke(message)
-    return context
+    return result
 
 def summarize_act(message, act_number):
     template = """" [INST] Ти асистент для надання відповідей з законодавства України.На вхід ти отримав один закон, підсуму його простими словами, викинь формальності та стандартні фрази. Додай усі зміни, які згадуються і цьому документі і опиши їх трьома реченнями. Якщо контекст пустий - пиши "Я не маю релевантної інформації. Спробуйте ще".
@@ -199,8 +199,8 @@ def summarize_act(message, act_number):
         {"context": ret2 , "question": RunnablePassthrough()}
     ).assign(answer=rag_chain_from_docs)
     context = ret2.get_relevant_documents(message)
-    #return rag_chain_summarize.invoke("")
-    return context
+    return rag_chain_summarize.invoke("")
+    #return context
 
 def generate_answer(message, history, checkbox, subject_input, rubric, date_beg, date_end, act_number):
     result = ""
@@ -222,8 +222,8 @@ def generate_answer(message, history, checkbox, subject_input, rubric, date_beg,
             result = generate_with_filters(message, subject_input, rubric, date_beg, date_end)
 
 
-    #result['answer'] =result['answer'].split("[/INST]")[-1].strip()
-    #formatted_results = format_result(result)
+    result['answer'] =result['answer'].split("[/INST]")[-1].strip()
+    formatted_results = format_result(result)
     return result