Spaces:

pseudolab
/

Balanced-News-Reading

Runtime error

gabrielwithhappy commited on Nov 5, 2023

Commit

1eb5133

1 Parent(s): d162cd8

remove top_p

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def get_summary(input_text):
     input_ids = tokenizer.encode(input_text, return_tensors="pt")
     summary_text_ids = summary_model.generate(
         length_penalty=1,
-        top_p=0.9,
         max_length=128,
         min_length=12,
         num_beams=4,

     input_ids = tokenizer.encode(input_text, return_tensors="pt")
     summary_text_ids = summary_model.generate(
         length_penalty=1,
+        # top_p=0.9,
         max_length=128,
         min_length=12,
         num_beams=4,