Update app.py

app.py

@@ -12,13 +12,13 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)
 
 
-nltk.download('punkt')
-nltk.download('brown')
-nltk.download('wordnet')
-nltk.download('omw-1.4')
+# nltk.download('punkt')
+# nltk.download('brown')
+# nltk.download('wordnet')
+# nltk.download('omw-1.4')
 
-from nltk.corpus import wordnet as wn
-from nltk.tokenize import sent_tokenize
+# from nltk.corpus import wordnet as wn
+# from nltk.tokenize import sent_tokenize
 
 def set_seed(seed: int):
     random.seed(seed)
@@ -28,12 +28,12 @@ def set_seed(seed: int):
 
 set_seed(42)
 
-def postprocesstext (content):
-  final=""
-  for sent in sent_tokenize(content):
-    sent = sent.capitalize()
-    final = final +" "+sent
-  return final
+# def postprocesstext (content):
+#   final=""
+#   for sent in sent_tokenize(content):
+#     sent = sent.capitalize()
+#     final = final +" "+sent
+#   return final
 
 def summarizer(text):
   text = text.strip().replace("\n"," ")
@@ -56,7 +56,7 @@ def summarizer(text):
 
   dec = [tokenizer.decode(ids,skip_special_tokens=True) for ids in outs]
   summary = dec[0]
-  summary = postprocesstext(summary)
+  #summary = postprocesstext(summary)
   summary= summary.strip()
 
   return summary