Update app.py

app.py

@@ -17,7 +17,7 @@ from string import punctuation
 # Prediction 
 def prediction(text):
   doc = nlp(text)
-  len1 = len(text.split())
+  len1 = len(text)
   tokens = [token.text for token in doc]
   word_frequencies = {}
   for word in doc:
@@ -41,7 +41,7 @@ def prediction(text):
           sentence_scores[sent] += word_frequencies[word.text.lower()]
   select_length = int(len(sentence_tokens)*0.3)
   summary = nlargest(select_length, sentence_scores, key = sentence_scores.get)
-  len2 = len(summary.split())
+  len2 = len(summary)
   return summary,len1,len2
 
 #text = """
@@ -64,8 +64,8 @@ def prediction(text):
 #print(summary)
 outputs = [
     gr.Textbox(lines =5,label = "Summarization of text"),
-    gr.Number(label="count1"),
-    gr.Number(label="count2")
+    gr.Number(label="Word Count of given Text"),
+    gr.Number(label="Word Count of Summarized Text")
 ]
 demo_app = gr.Interface(
     fn=prediction,