Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,6 +4,10 @@ import nltk
 import math
 import torch
 
+
+nltk.download('punkt')
+nltk.download('punkt_tab')
+
 model_name = "Timur1984/t5-base-title-generation"
 max_input_length = 512
 
@@ -16,8 +20,6 @@ def load_model():
     print("Loading model...")
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-    nltk.download('punkt')
-    nltk.download('punkt_tab')
     print("Model loaded!")
     return tokenizer, model