Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,12 +16,18 @@ st_model_load = st.text('Loading title generator model...')
 
 @st.cache_resource
 def load_model():
-    print("Loading model...")
-    tokenizer = T5Tokenizer.from_pretrained(model_name, trust_remote_code=True)
-    model = T5ForConditionalGeneration.from_pretrained(model_name, trust_remote_code=True)
-    nltk.download('punkt')
-    print("Model loaded!")
+    try:
+        print("Loading model...")
+        tokenizer = T5Tokenizer.from_pretrained(model_name, trust_remote_code=True)
+        model = T5ForConditionalGeneration.from_pretrained(model_name, trust_remote_code=True)
+    except ValueError as e:
+        st.error(f"Full error message: {str(e)}")
+        raise
+        
+        nltk.download('punkt')
+        print("Model loaded!")
     return tokenizer, model
+    
 
 tokenizer, model = load_model()
 st.success('Model loaded!')