add reqments & max length

ebart.py

@@ -55,8 +55,11 @@ class PegasusSummarizer:
 
 @spaces.GPU
 def generate_summary(text, max_length=180, min_length=64):
-    summarizer = PegasusSummarizer()
-    return summarizer.generate_summary(text, max_length, min_length)
+    if(len(text)< max_length):
+        return text
+    else:
+        summarizer = PegasusSummarizer()
+        return summarizer.generate_summary(text, max_length, min_length)
 
 if __name__ == "__main__":
     text = (

requirements.txt

@@ -2,3 +2,4 @@ transformers
 torch
 scikit-learn
 pandas
+tokenizers_pegasus