commit

app.py

@@ -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,
+        pegasus_tokenizer(text,return_tensors="pt").input_ids,
         max_length=max_length,
         num_beams=num_beams,
         early_stopping=early_stopping