Spaces:
Sleeping
Sleeping
File size: 476 Bytes
ca9a177 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from google import SemanticSearch
from transformers import AutoTokenizer, AutoModel
class RAGModel:
def __init__(self, configs) -> None:
self.configs = configs
model_url = configs["RAG"]["genration_model"]
self.model = AutoModel.from_pretrained(model_url)
self.tokenizer = AutoTokenizer.from_pretrained(model_url)
def create_propmt(self, topk_items: list[str]):
def answer_query(self, query: str, context: list[str]) :
|