Update app.py

app.py

@@ -2,18 +2,18 @@ import gradio as gr
 from transformers import pipeline
 
 # Modelos geradores otimizados
-model_a = pipeline("text-generation", model="gpt2", return_full_text=False)
+model_a = pipeline("text-generation", model="Canarim-Instruct-PTBR-Dataset", return_full_text=False)
 model_b = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", return_full_text=False)
 
 # Modelo juiz
-arbiter = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
+arbiter = pipeline("text-classification", model="neuralmind/bert-base-portuguese-cased")
 
 # Julgamento por batch
 def judge_response(response_a, response_b):
     results = arbiter([response_a, response_b])
     score_a = results[0]['score']
     score_b = results[1]['score']
-    return ("Model A", response_a) if score_a > score_b else ("Model B", response_b)
+    return ("Modelo A", response_a) if score_a > score_b else ("Modelo B", response_b)
 
 # Função principal
 def chatbot(prompt):
@@ -28,7 +28,7 @@ iface = gr.Interface(
     inputs=gr.Textbox(label="Digite sua pergunta:"),
     outputs=[
         gr.Textbox(label="Pergunta"),
-        gr.Textbox(label="Resposta do Modelo A (GPT-2)"),
+        gr.Textbox(label="Resposta do Modelo A (Canarim-Instruct-PTBR-Dataset)"),
         gr.Textbox(label="Resposta do Modelo B (GPT-Neo-125M)"),
         gr.Textbox(label="Modelo Vencedor"),
         gr.Textbox(label="Resposta Escolhida"),