Update app.py

app.py

@@ -52,9 +52,9 @@ def translate(text, model, tokenizer, source_lang="pt", target_lang="en"):
     return translated_text
 
 # Function to translate and interact with TAPEX model
-def response(user_question, table_data):
+def response(user_question, all_anomalies):
     question_en = translate(user_question, pt_en_translator, tokenizer, source_lang="pt", target_lang="en")
-    encoding = tapex_tokenizer(table=table_data, query=[question_en], padding=True, return_tensors="pt", truncation=True)
+    encoding = tapex_tokenizer(table=all_anomalies, query=[question_en], padding=True, return_tensors="pt", truncation=True)
     outputs = tapex_model.generate(**encoding)
     response_en = tapex_tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
     response_pt = translate(response_en, en_pt_translator, tokenizer, source_lang="en", target_lang="pt")
@@ -166,6 +166,8 @@ def apply_prophet(df_clean):
     # Return the dataframe of all anomalies
     st.write(f"Concluída a aplicação do modelo de série tempotal")
     st.write(all_anomalies.head())
+    all_anomalies.sort_values(by=['real'], ascending=False, inplace=True)
+    all_anomalies = all_anomalies[all_anomalies['real'] >= 10,000,000.00]
     return all_anomalies
 
 # Initialize session states