Kokoro-API-4

Sleeping

yaron123 commited on Jan 19

Commit

9d7dc73

1 Parent(s): 5cb26b4

commit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -127,7 +127,7 @@ pegasus_model = PegasusForConditionalGeneration.from_pretrained(pegasus_name)
 def summarize_text(text, model_name="google/pegasus-xsum", max_length=30, num_beams=4, early_stopping=True):
     return pegasus_tokenizer.decode( pegasus_model.generate(
-        pegasus_tokenizer(pos_lyrics,return_tensors="pt").input_ids,
         max_length=max_length,
         num_beams=num_beams,
         early_stopping=early_stopping

 def summarize_text(text, model_name="google/pegasus-xsum", max_length=30, num_beams=4, early_stopping=True):
     return pegasus_tokenizer.decode( pegasus_model.generate(
+        pegasus_tokenizer(text,return_tensors="pt").input_ids,
         max_length=max_length,
         num_beams=num_beams,
         early_stopping=early_stopping