Merge branch 'main' of https://huggingface.co/spaces/EE21/ToS-Summarization

keyword_extraction.py

@@ -3,6 +3,7 @@ from rake_nltk import Rake
 
 # Download NLTK data (if not already downloaded)
 nltk.download('punkt')
+nltk.download('stopwords')
 
 def extract_keywords(text):
     # Initialize Rake with stopwords set to None (to keep all words)
@@ -14,5 +15,4 @@ def extract_keywords(text):
     # Get the ranked keywords
     ranked_keywords = rake.get_ranked_phrases()
     
-    return ranked_keywords
-
+    return ranked_keywords