Spaces:
Sleeping
Sleeping
Update rag_utils.py
Browse files- rag_utils.py +1 -1
rag_utils.py
CHANGED
@@ -25,7 +25,7 @@ def query_index(question, index, documents, model, k=3):
|
|
25 |
|
26 |
def generate_answer(question, context):
|
27 |
token = os.getenv("HUGGINGFACE") # requis pour Mistral
|
28 |
-
model_id = "
|
29 |
|
30 |
tokenizer = AutoTokenizer.from_pretrained(model_id, token=token)
|
31 |
tokenizer.pad_token = tokenizer.eos_token
|
|
|
25 |
|
26 |
def generate_answer(question, context):
|
27 |
token = os.getenv("HUGGINGFACE") # requis pour Mistral
|
28 |
+
model_id = "google/flan-t5-base"
|
29 |
|
30 |
tokenizer = AutoTokenizer.from_pretrained(model_id, token=token)
|
31 |
tokenizer.pad_token = tokenizer.eos_token
|