Spaces:

0edon
/

KairosNews

Sleeping

Quintino Fernandes commited on May 17

Commit

a897978

1 Parent(s): da25e0f

better model parameters

Files changed (1) hide show

models/summarization.py CHANGED Viewed

@@ -21,8 +21,8 @@ class SummarizationModel:
             inputs,
             max_length=512,
             min_length=128,
-            num_beams=3,
-            no_repeat_ngram_size=4,
             early_stopping=False,
         )

             inputs,
             max_length=512,
             min_length=128,
+            num_beams=5,
+            no_repeat_ngram_size=3,
             early_stopping=False,
         )