Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import nltk
 import math
 import torch
-# nltk.download('punkt_tab')
+nltk.download('punkt_tab')
 nltk.download('punkt')
 
 
@@ -19,7 +19,7 @@ def load_model():
     print("Loading model...")
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-    nltk.download('punkt')
+    # nltk.download('punkt')
     print("Model loaded!")
     return tokenizer, model