Spaces:
Sleeping
Sleeping
ping98k
commited on
Commit
·
1633c5a
1
Parent(s):
1d9f28c
Update main.py
Browse files
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"),
|