Update app.py

app.py

@@ -6,7 +6,7 @@ model_a = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-0528", trus
 model_b = pipeline("text-generation", model="botbot-ai/CabraLlama3-8b")
 
 # Modelo juiz
-arbiter = pipeline("text-classification", model="neuralmind/bert-base-portuguese-cased", trust_remote_code=True)
+arbiter = pipeline("text-classification", model="neuralmind/bert-base-portuguese-cased")
 
 # Julgamento por batch
 def judge_response(response_a, response_b):
@@ -17,8 +17,11 @@ def judge_response(response_a, response_b):
 
 # Função principal
 def chatbot(prompt):
+    # Gera respostas para ambos os modelos
     response_a = model_a(prompt, max_new_tokens=40)[0]['generated_text'].strip()
     response_b = model_b(prompt, max_new_tokens=40)[0]['generated_text'].strip()
+    
+    # Julga as respostas
     winner, final_response = judge_response(response_a, response_b)
     return prompt, response_a, response_b, winner, final_response