Add nltk punct

README.md

@@ -1,6 +1,6 @@
 ---
 title: Emotion Detection
-emoji: 🐠
+emoji: 📝➡️🚀
 colorFrom: blue
 colorTo: yellow
 sdk: docker

extract.py

@@ -1,5 +1,5 @@
-# import nltk
-# nltk.download('punkt')
+import nltk
+nltk.download('punkt')
 
 from nltk.cluster.util import cosine_distance
 from nltk.tokenize import sent_tokenize, word_tokenize
@@ -54,7 +54,7 @@ def get_pagerank(importance, top_k):
 
 
 def summarize_text(text: str) -> str:
-    return text
+    # return text
     sentences = sent_tokenize(text)[:2000]
     top_k = max(20, int(len(sentences) ** .5))
     mat = build_matrix(sentences)

inference.py

@@ -27,6 +27,7 @@ def predict_emotions(text: str) -> str:
     return '\n'.join(f"{k}: {v}%" for k, v in sorted(emotions_list.items(),
                                                     key=lambda x: -x[1]))
 
+# path_gram = 'Djacon/mbert-gram'
 path_gram = './models/mbert-gram/'
 model_gram = BertForSequenceClassification.from_pretrained(path_gram)
 tokenizer_gram = AutoTokenizer.from_pretrained(path_gram)