Feelings_to_Emoji / config.py
Dan Mo
Add script to generate and save embeddings for models
cfb0d15
raw
history blame contribute delete
827 Bytes
"""
Configuration settings for the Emoji Mashup application.
"""
# Application configuration
CONFIG = {
"model_name": "all-mpnet-base-v2",
"emotion_file": "google-emoji-kitchen-emotion.txt",
"item_file": "google-emoji-kitchen-item.txt",
"emoji_kitchen_url": "https://emojik.vercel.app/s/{emoji1}_{emoji2}",
"default_size": 256
}
# Available embedding models
EMBEDDING_MODELS = {
"mpnet": {
"id": "all-mpnet-base-v2",
"size": "110M",
"notes": "Balanced, great general-purpose model"
},
"gte": {
"id": "thenlper/gte-large",
"size": "335M",
"notes": "Context-rich, good for emotion & nuance"
},
"bge": {
"id": "BAAI/bge-large-en-v1.5",
"size": "350M",
"notes": "Tuned for ranking & high-precision similarity"
}
}