Subha95 commited on
Commit
4356a00
·
verified ·
1 Parent(s): 3333075

Update chatbot_rag.py

Browse files
Files changed (1) hide show
  1. chatbot_rag.py +1 -1
chatbot_rag.py CHANGED
@@ -35,7 +35,7 @@ def build_qa():
35
  print("✅ LLM loaded.")
36
 
37
  pipe = pipeline(
38
- "text-generation",
39
  model=model,
40
  tokenizer=tokenizer,
41
  max_new_tokens=256,
 
35
  print("✅ LLM loaded.")
36
 
37
  pipe = pipeline(
38
+ "question-answering",
39
  model=model,
40
  tokenizer=tokenizer,
41
  max_new_tokens=256,