Update app.py

app.py

@@ -22,6 +22,10 @@ model.to(device)
 # model_name = "roberta-base"
 # tokenizer = RobertaTokenizer.from_pretrained(model_name, map_location=torch.device('cpu'))
 
+def count_words(text):
+    words = text.split()  # Split the text into a list of words
+    return len(words)
+    
 def text_to_sentences(text):
     clean_text = text.replace('\n', ' ')
     return re.split(r'(?<=[^A-Z].[.?]) +(?=[A-Z])', clean_text)