Spaces:

Leesn465
/

fastapi-stock-api

Sleeping

Leesn465 commited on Jan 10

Commit

d27d3eb

verified ·

1 Parent(s): b05f4d5

Update util/keywordExtract.py

Files changed (1) hide show

util/keywordExtract.py CHANGED Viewed

@@ -27,7 +27,7 @@ def summarize_kobart(text):
         text,
         return_tensors="pt",
         truncation=True,
-        max_length=1024,  # 모델에 맞게 조정 (512/1024 중 하나일 확률 큼)
     )
     summary_ids = summary_model.generate(

         text,
         return_tensors="pt",
         truncation=True,
+        max_length=512,  # 모델에 맞게 조정 (512/1024 중 하나일 확률 큼)
     )
     summary_ids = summary_model.generate(