Update src/streamlit_app.py

src/streamlit_app.py

@@ -11,10 +11,15 @@ def load_tokenizer():
 # ✅ Load model
 @st.cache_resource
 def load_model():
-    model_path = "final_injury_model.pt"  # or use a Hugging Face model ID if from hub
+    model_path = "model/final_injury_model.pt"  # or use a Hugging Face model ID if from hub
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = torch.load(model_path, map_location=device)
+    if not os.path.exists(model_path):
+    st.error(f"Model file not found at: {model_path}")
+    return None
     model.eval()
+ 
+
     return model
 def main():
     st.title("NBA Injury Type & Duration Classifier 🏀")