Spaces:
Sleeping
Sleeping
from models.model_seeds import seeds, seed2str | |
# ISSUES = ['Anxiety','Suicide'] | |
ISSUES = [k for k,_ in seeds.items()] | |
SOURCES = [ | |
# "CTL_llama2", | |
"CTL_llama3", | |
# "CTL_mistral", | |
'OA_rolemodel', | |
# 'OA_finetuned', | |
] | |
SOURCES_LAB = {"OA_rolemodel":'OpenAI GPT3.5', | |
"OA_finetuned":'Finetuned OpenAI', | |
# "CTL_llama2": "Llama 2", | |
"CTL_llama3": "Llama 3", | |
"CTL_mistral": "Mistral", | |
} | |
ENDPOINT_NAMES = { | |
# "CTL_llama2": "texter_simulator", | |
"CTL_llama3": "texter_simulator_llm", | |
# 'CTL_llama2': "llama2_convo_sim", | |
"CTL_mistral": "convo_sim_mistral" | |
} | |
def source2label(source): | |
return SOURCES_LAB[source] | |
def issue2label(issue): | |
return seed2str.get(issue, "GCT") | |
ENVIRON = "dev" | |
DB_SCHEMA = 'prod_db' if ENVIRON == 'prod' else 'test_db' | |
DB_CONVOS = 'conversations' | |
DB_COMPLETIONS = 'comparison_completions' | |
DB_BATTLES = 'battles' | |
DB_ERRORS = 'completion_errors' |