minor

app.py

@@ -113,7 +113,7 @@ class EntailmentChecker(BaseComponent):
             "aggregate_entailment_info": aggregate_entailment_info,
         }
 
-        return entailment_checker_result
+        return entailment_checker_result, "output_1"
     
     def run_batch(self, queries: List[str], documents: List[Document]):
         entailment_checker_result_batch = []
@@ -177,11 +177,7 @@ def start_haystack():
 
     return pipe
 
-def run_statement(statement: str, retriever_top_k: int = 5):
-    global pipe 
-    pipe = start_haystack()
-    run = check_statement(statement, retriever_top_k)
-    return run
+pipe = start_haystack()
 
 @st.cache_resource
 def check_statement(statement: str, retriever_top_k: int = 5):
@@ -256,7 +252,7 @@ def main():
         st.session_state.statement = statement
         with st.spinner("   Procurando a Similaridade no banco de sentenças..."):
             try:
-                st.session_state.results = run_statement(statement, RETRIEVER_TOP_K)
+                st.session_state.results = check_statement(statement, RETRIEVER_TOP_K)
                 print(f"S: {statement}")
                 time_end = time.time()
                 print(time.strftime("%Y-%m-%d %H:%M:%S", time.gmtime()))