File size: 411 Bytes
bb4e28f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
MODEL_CONFIG = {
    "llm": {
        "provider": "groq",
        "model": "mistral-7b-instruct",
    },
    "embedding_model": {
        "name": "all-MiniLM-L6-v2",
        "provider": "sentence-transformers"
    },
    "summarizer": {
        "use_groq": True,
        "fallback_model": "TinyLlama"
    },
    "sentiment_model": {
        "name": "ProsusAI/finbert",
        "provider": "huggingface"
    }
}