Update app.py

app.py

@@ -69,11 +69,17 @@ def make_summary(word):
     # Apply get_sentence_embeddings to dataframe to create column Embeddings
     data['Embeddings'] = get_sentence_embeddings()
     
-    #Number of expected sentences
-    if len(tokens) <= 4: 
+    #Number of expected sentences for shorter summaries
+    #if len(tokens) <= 4: 
+       # NUM_CLUSTERS = 1
+    #else:
+        #NUM_CLUSTERS = len(tokens)//4
+        
+      #Number of expected sentences for medium summaries
+    if len(tokens) <= 1: 
         NUM_CLUSTERS = 1
     else:
-        NUM_CLUSTERS = len(tokens)//4
+        NUM_CLUSTERS = len(tokens)//2
 
     iterations = 25
     # Convert Embeddings into an array and store in variable X