File size: 373 Bytes
5f0df75 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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" |