Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,10 +16,13 @@ import streamlit as st
 import torch
 from transformers import AutoTokenizer, AutoModelForTokenClassification
 from dotenv import load_dotenv
+from pathlib import Path
 load_dotenv()
 
 import base64
-
+# Absolute path to this file's directory (e.g., /app/src)
+HERE = Path(__file__).resolve().parent
+ASSETS_DIR = HERE / "assets"
 APP_HOME_URL = "https://www.brundagelab.org/research/apps/"  # change to your desired destination
 
 def _img_to_data_uri(path: str) -> str:
@@ -33,7 +36,7 @@ def brundage_header():
     col1, col2 = st.columns([2, 5])
 
     with col1:
-        logo_path = "assets/brundage_logo.png"
+        logo_path = ASSETS_DIR / "brundage_logo.png"
         logo_uri = _img_to_data_uri(logo_path)
 
         st.markdown(