Upload app.py

app.py

@@ -41,7 +41,13 @@ def get_wiki_summary(inp):
             if lemmatizer.lemmatize(str(wrd).lower()) in word_counts.keys():
                 sentence_scores[sentence] += word_counts[lemmatizer.lemmatize(str(wrd).lower())]
 
-    summary_length = int(len(sentence_scores)*0.20)
+    summary_length = 0
+
+    if len(sentence_scores) > 5 :
+        summary_length = int(len(sentence_scores)*0.20)
+    else:
+        summary_length = int(len(sentence_scores)*0.50)
+        
     summary = str()
 
     for sentence in nltk.sent_tokenize(text):