minko186 commited on
Commit
81f38b2
1 Parent(s): bfcf637

Create config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +13 -0
config.yaml ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ QUILLBOT_LABELS : ["Original", "Altered"]
2
+ MC_OUTPUT_LABELS : ["Claude", "Gemini", "LLaMA3", "Mistral", "OpenAI GPT"]
3
+ 1ON1_OUTPUT_LABELS : ['OpenAI GPT', 'Mistral', 'CLAUDE', 'Gemini', 'Grammar Enhancer']
4
+ TEXT_MC_MODEL_PATH : 'polygraf-ai/mc-model-contrastive-loss'
5
+ TEXT_BC_MODEL_PATH : 'polygraf-ai/bc-roberta-openai-4mcomp-512-lighterstop'
6
+ BIAS_CHECKER_MODEL_PATH : "textattack/roberta-base-CoLA"
7
+ BIAS_CORRECTOR_MODEL_PATH : "pszemraj/flan-t5-large-grammar-synthesis"
8
+ TEXT_QUILLBOT_MODEL_PATH : 'polygraf-ai/humanizer-model'
9
+ TEXT_1ON1_MODEL: ['polygraf-ai/gpt4-1on1-v2', 'polygraf-ai/mistral_ai-1on1-v2', 'polygraf-ai/claude-1on1-v2', 'polygraf-ai/bard-1on1-v2', 'polygraf-ai/llama2-1on1-v2']
10
+ READABILITY_MODEL_ID : 'openai-community/gpt2-medium'
11
+ MC_TOKEN_SIZE : '512'
12
+ BC_TOKEN_SIZE : '333'
13
+ ASSEMBLY_AI_TOKEN : '698d941a5e39402b8e9009979b37a520'