Programmes commited on
Commit
940fe8c
·
verified ·
1 Parent(s): 21a4e1a

Update rag_utils.py

Browse files
Files changed (1) hide show
  1. 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 = "mgoogle/flan-t5-base"
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