nanfangwuyu21's picture
change summary model to openai gpt-4o-mini
1f16680
raw
history blame contribute delete
407 Bytes
# 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")