Caikejs commited on
Commit
5b14e76
·
verified ·
1 Parent(s): ba91276

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -2,11 +2,11 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Modelos geradores otimizados
5
- model_a = pipeline("text-generation", model="Canarim-Instruct-PTBR-Dataset", return_full_text=False)
6
- model_b = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", return_full_text=False)
7
 
8
  # Modelo juiz
9
- arbiter = pipeline("text-classification", model="neuralmind/bert-base-portuguese-cased")
10
 
11
  # Julgamento por batch
12
  def judge_response(response_a, response_b):
@@ -28,8 +28,8 @@ iface = gr.Interface(
28
  inputs=gr.Textbox(label="Digite sua pergunta:"),
29
  outputs=[
30
  gr.Textbox(label="Pergunta"),
31
- gr.Textbox(label="Resposta do Modelo A (Canarim-Instruct-PTBR-Dataset)"),
32
- gr.Textbox(label="Resposta do Modelo B (GPT-Neo-125M)"),
33
  gr.Textbox(label="Modelo Vencedor"),
34
  gr.Textbox(label="Resposta Escolhida"),
35
  ],
 
2
  from transformers import pipeline
3
 
4
  # Modelos geradores otimizados
5
+ model_a = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-0528", trust_remote_code=True)
6
+ model_b = pipeline("text-generation", model="botbot-ai/CabraLlama3-8b")
7
 
8
  # Modelo juiz
9
+ arbiter = pipeline("text-classification", model="neuralmind/bert-base-portuguese-cased", trust_remote_code=True)
10
 
11
  # Julgamento por batch
12
  def judge_response(response_a, response_b):
 
28
  inputs=gr.Textbox(label="Digite sua pergunta:"),
29
  outputs=[
30
  gr.Textbox(label="Pergunta"),
31
+ gr.Textbox(label="Resposta do Modelo A (DeepSeek-R1-0528)"),
32
+ gr.Textbox(label="Resposta do Modelo B (CabraLlama3-8b)"),
33
  gr.Textbox(label="Modelo Vencedor"),
34
  gr.Textbox(label="Resposta Escolhida"),
35
  ],