Spaces:

Rahmat82
/

RHM-text-summarizer-light

Sleeping

Rahmat82 commited on Mar 4, 2024

Commit

47c8605

verified ·

1 Parent(s): 2c78b7a

set max_new_tokens to 128

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ async def summarize(text, retries=6):
     data = {
         "inputs": text,
         "options": {
-            "max_new_tokens": 100,
             "min_length": 10,
             "max_length": 500,
             "length_penalty": 2.0,

     data = {
         "inputs": text,
         "options": {
+            "max_new_tokens": 128,
             "min_length": 10,
             "max_length": 500,
             "length_penalty": 2.0,