Update app.py

app.py

@@ -11,50 +11,55 @@ def query(payload):
 
 
 def summarize(input_text):
-    max_chunk = 500
+    lister= input_text.split(" ")
+    word_lenght = len(lister)
+    if word_lenght < 50:
+        text= "Oops! It looks like your sentence is too short to summarize."
+    else:
+        max_chunk = 500
 
-    input_text = input_text.replace('.', '.<eos>')
+        input_text = input_text.replace('.', '.<eos>')
 
-    input_text = input_text.replace('?', '?<eos>')
+        input_text = input_text.replace('?', '?<eos>')
 
-    input_text = input_text.replace('!', '!<eos>')
+        input_text = input_text.replace('!', '!<eos>')
 
-    sentences = input_text.split('<eos>')
+        sentences = input_text.split('<eos>')
 
-    current_chunk = 0
+        current_chunk = 0
 
-    chunks = []
+        chunks = []
 
-    for sentence in sentences:
+        for sentence in sentences:
 
-        if len(chunks) == current_chunk + 1:
+            if len(chunks) == current_chunk + 1:
 
-            if len(chunks[current_chunk]) + len(sentence.split(' ')) <= max_chunk:
+                if len(chunks[current_chunk]) + len(sentence.split(' ')) <= max_chunk:
 
-                chunks[current_chunk].extend(sentence.split(' '))
+                    chunks[current_chunk].extend(sentence.split(' '))
 
-            else:
+                else:
 
-                current_chunk += 1
+                    current_chunk += 1
 
-                chunks.append(sentence.split(' '))
+                    chunks.append(sentence.split(' '))
 
-        else:
+            else:
 
-            print(current_chunk)
+                print(current_chunk)
 
-            chunks.append(sentence.split(' '))
+                chunks.append(sentence.split(' '))
 
 
-    for chunk_id in range(len(chunks)):
+        for chunk_id in range(len(chunks)):
 
-        chunks[chunk_id] = ' '.join(chunks[chunk_id])
+            chunks[chunk_id] = ' '.join(chunks[chunk_id])
 
 
-    res = query({"inputs":chunks})
+        res = query({"inputs":chunks})
 
 
-    text = ' '.join([summ['summary_text'] for summ in res])
+        text = ' '.join([summ['summary_text'] for summ in res])
 
     return text