config.py

AVAILABLE_MODELS = {
    "BERT" : 'cancerfarore/bert-base-uncased-CancerFarore-Model',
    "ALBERT" : "cancerfarore/albert-base-v2-CancerFarore-Model",
    "ROBERTA" : "cancerfarore/roberta-base-CancerFarore-Modela",
    "PACIFISTA" : "cancerfarore/roberta-base-CancerFarore-Model"
}

MODEL_NAMES = []
for NAME in AVAILABLE_MODELS:
    MODEL_NAMES.append(NAME)

TITRE  = "Pacifista 🐻"

# API_URL = 'http://127.0.0.1:5000'
API_URL = 'http://52.4.151.236:5000'
# API_URL = 'http://10.27.86.198:5000'
USER_AVATAR = "👤"
BOT_AVATAR = "🤖"