File size: 407 Bytes
868a18b
f0fcc0f
 
 
2246808
f0fcc0f
 
1f16680
 
f0fcc0f
1
2
3
4
5
6
7
8
9
10
# from app.models.tinyllama import TinyLlamaModel
from app.models.gpt4omini import GPT4OMini
from app.models.bart_large_cnn import BartSummaryModel
from langchain_huggingface import HuggingFaceEmbeddings

# model = TinyLlamaModel()
LLM = GPT4OMini()
# Summary_Model = BartSummaryModel()
Summary_Model = GPT4OMini()
Embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")