| from conversation import Conversation_RAG | |
| from vector_index import * | |
| class ModelSetup: | |
| def __init__(self, model_name): | |
| self.model_name = model_name | |
| def setup(self): | |
| conv_rag = Conversation_RAG(self.model_name) | |
| self.vectordb = conv_rag.get_vectordb() | |
| self.pipeline = conv_rag.create_model() | |
| return "Model Setup Complete" |