Spaces:

bshk57
/

FlaskAsk

Sleeping

bshk57 commited on Mar 1

Commit

1e82666

verified ·

1 Parent(s): 5a1ac75

update in line 314

Files changed (1) hide show

app_new.py CHANGED Viewed

@@ -311,7 +311,7 @@ def initialize_model():
     print("\n🤖 Loading LLM...")
     generator = pipeline(
-        "text2text-generation",
         model=LLM_MODEL,
         tokenizer=LLM_MODEL,
         max_new_tokens=200,            # Reduced from 300

     print("\n🤖 Loading LLM...")
     generator = pipeline(
+        "text-generation",
         model=LLM_MODEL,
         tokenizer=LLM_MODEL,
         max_new_tokens=200,            # Reduced from 300