Update app.py

app.py

@@ -81,8 +81,10 @@ model_path = "./saved_model"
 tokenizer_path = "./saved_tokenizer"  # Define this path for saved tokenizer
 
 try:
+    # Check if the tokenizer is saved, if not, add custom tokens
     if not os.path.exists(tokenizer_path):
         tokenizer = T5Tokenizer.from_pretrained("t5-small")
+        tokenizer.add_tokens(['<extra_id_99>'])  # Add custom token if required
         tokenizer.save_pretrained(tokenizer_path)
     else:
         tokenizer = T5Tokenizer.from_pretrained(tokenizer_path, local_files_only=True)
@@ -142,4 +144,3 @@ st.markdown("""
     Smart Text Summarizer - Crafted with hadheedo
 </div>
 """, unsafe_allow_html=True)
-