Spaces:

bshk57
/

FlaskAsk

Sleeping

bshk57 commited on Mar 1

Commit

ec8675e

verified ·

1 Parent(s): ffc9415

text2text -> text in line 160

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(
-        "text2text-generation",
         model=LLM_MODEL,
         tokenizer=LLM_MODEL,
         max_new_tokens=200,

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