Spaces:

bshk57
/

FlaskAsk

Runtime error

bshk57 commited on 9 days ago

Commit

ffc9415

verified ·

1 Parent(s): c78c5ec

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -157,7 +157,7 @@ def initialize_model():
     retriever = vectordb.as_retriever(search_kwargs={"k": 4})
     generator = pipeline(
-        "text-generation",
         model=LLM_MODEL,
         tokenizer=LLM_MODEL,
         max_new_tokens=200,

     retriever = vectordb.as_retriever(search_kwargs={"k": 4})
     generator = pipeline(
+        "text2text-generation",
         model=LLM_MODEL,
         tokenizer=LLM_MODEL,
         max_new_tokens=200,