ping98k commited on
Commit
1633c5a
·
1 Parent(s): 1d9f28c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -17,7 +17,7 @@ PAIRWISE_FILTER_DEFAULT = os.getenv("ENABLE_PAIRWISE_FILTER", "true").lower() ==
17
  GENERATE_MODEL_DEFAULT = os.getenv("GENERATE_MODEL", "gpt-4o-mini")
18
  SCORE_MODEL_DEFAULT = os.getenv("SCORE_MODEL", "gpt-4o-mini")
19
  PAIRWISE_MODEL_DEFAULT = os.getenv("PAIRWISE_MODEL", "gpt-4o-mini")
20
-
21
  def _clean_json(txt):
22
  txt = re.sub(r"^```.*?\n|```$", "", txt, flags=re.DOTALL).strip()
23
  try:
@@ -234,7 +234,7 @@ demo = gr.Interface(
234
  gr.Textbox(value=SCORE_MODEL_DEFAULT, label="Score Model"),
235
  gr.Textbox(value=PAIRWISE_MODEL_DEFAULT, label="Pairwise Model"),
236
  gr.Textbox(lines=10, label="Instruction"),
237
- gr.Textbox(lines=5, label="Criteria (comma separated)"),
238
  gr.Number(value=NUM_GENERATIONS_DEFAULT, label="Number of Generations"),
239
  gr.Number(value=POOL_SIZE_DEFAULT, label="Top Picks Score Filter"),
240
  gr.Number(value=NUM_TOP_PICKS_DEFAULT, label="Top Picks Pairwise"),
 
17
  GENERATE_MODEL_DEFAULT = os.getenv("GENERATE_MODEL", "gpt-4o-mini")
18
  SCORE_MODEL_DEFAULT = os.getenv("SCORE_MODEL", "gpt-4o-mini")
19
  PAIRWISE_MODEL_DEFAULT = os.getenv("PAIRWISE_MODEL", "gpt-4o-mini")
20
+ CRITERIA_DEFAULT = "Factuality,Instruction Following,Precision"
21
  def _clean_json(txt):
22
  txt = re.sub(r"^```.*?\n|```$", "", txt, flags=re.DOTALL).strip()
23
  try:
 
234
  gr.Textbox(value=SCORE_MODEL_DEFAULT, label="Score Model"),
235
  gr.Textbox(value=PAIRWISE_MODEL_DEFAULT, label="Pairwise Model"),
236
  gr.Textbox(lines=10, label="Instruction"),
237
+ gr.Textbox(value=CRITERIA_DEFAULT, lines=5, label="Criteria (comma separated)"),
238
  gr.Number(value=NUM_GENERATIONS_DEFAULT, label="Number of Generations"),
239
  gr.Number(value=POOL_SIZE_DEFAULT, label="Top Picks Score Filter"),
240
  gr.Number(value=NUM_TOP_PICKS_DEFAULT, label="Top Picks Pairwise"),