Caikejs commited on
Commit
ed42ce1
·
verified ·
1 Parent(s): b8715d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ model_a = pipeline("text-generation", model="distilgpt2") # Modelo A
6
  model_b = pipeline("text-generation", model="EleutherAI/gpt-neo-125M") # Modelo B
7
 
8
  # Modelo juiz
9
- arbiter = pipeline("text-classification", model="neuralmind/bert-base-portuguese-cased") # Modelo Árbitro
10
 
11
  # Julgamento por batch
12
  def judge_response(response_a, response_b):
 
6
  model_b = pipeline("text-generation", model="EleutherAI/gpt-neo-125M") # Modelo B
7
 
8
  # Modelo juiz
9
+ arbiter = pipeline("text-classification",model="google/flan-t5-small") # Modelo Árbitro
10
 
11
  # Julgamento por batch
12
  def judge_response(response_a, response_b):