Spaces:

Subha95
/

Harry_potter_wiki

Sleeping

Subha95 commited on Sep 3, 2025

Commit

436e90f

verified ·

1 Parent(s): 5bafd9a

Update chatbot_rag.py

Files changed (1) hide show

chatbot_rag.py CHANGED Viewed

@@ -34,7 +34,7 @@ def build_qa():
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, device_map="auto")
     pipe = pipeline(
-        "text-generation",
         model=model,
         tokenizer=tokenizer,
         max_new_tokens=300,

     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, device_map="auto")
     pipe = pipeline(
+        "text2text-generation",
         model=model,
         tokenizer=tokenizer,
         max_new_tokens=300,