Spaces:
Sleeping
Sleeping
File size: 607 Bytes
59d5667 75b7dbb 975a927 75b7dbb 975a927 b3eeb77 975a927 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from models.model_seeds import seeds
# ISSUES = ['Anxiety','Suicide']
ISSUES = [k for k,_ in seeds.items()]
SOURCES = ['OA_rolemodel',
# 'OA_finetuned',
"CTL_llama2"]
SOURCES_LAB = {"OA_rolemodel":'OpenAI GPT3.5',
"OA_finetuned":'Finetuned OpenAI',
"CTL_llama2": "Custom CTL"
}
def source2label(source):
return SOURCES_LAB[source]
ENVIRON = "prod"
DB_SCHEMA = 'prod_db' if ENVIRON == 'prod' else 'test_db'
DB_CONVOS = 'conversations'
DB_COMPLETIONS = 'comparison_completions'
DB_BATTLES = 'battles'
DB_ERRORS = 'completion_errors' |