copyright_checker / config.yaml
aliasgerovs's picture
Update config.yaml
138131d verified
raw
history blame
835 Bytes
QUILLBOT_LABELS : ["Original", "Altered"]
MC_OUTPUT_LABELS : ["Claude", "Gemini", "LLaMA3", "Mistral", "OpenAI GPT"]
1ON1_OUTPUT_LABELS : ['OpenAI GPT', 'Mistral', 'CLAUDE', 'Gemini', 'Grammar Enhancer']
TEXT_MC_MODEL_PATH : 'polygraf-ai/mc-model-contrastive-loss'
TEXT_BC_MODEL_PATH : 'polygraf-ai/bc-model-finetuned-government'
BIAS_CHECKER_MODEL_PATH : "textattack/roberta-base-CoLA"
BIAS_CORRECTOR_MODEL_PATH : "pszemraj/flan-t5-large-grammar-synthesis"
TEXT_QUILLBOT_MODEL_PATH : 'polygraf-ai/humanizer-model'
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']
READABILITY_MODEL_ID : 'openai-community/gpt2-medium'
MC_TOKEN_SIZE : '512'
BC_TOKEN_SIZE : '333'
ASSEMBLY_AI_TOKEN : '698d941a5e39402b8e9009979b37a520'