Update src/streamlit_app.py
Browse files- src/streamlit_app.py +4 -3
src/streamlit_app.py
CHANGED
@@ -74,9 +74,7 @@ class RichGCNModel(nn.Module):
|
|
74 |
# ------------------- UI Setup -------------------
|
75 |
st.set_page_config(layout="wide", page_title="Drug Toxicity Predictor")
|
76 |
st.title("π§ͺ Drug Toxicity Prediction Dashboard")
|
77 |
-
|
78 |
-
st.markdown(f"π **GCN Model Loaded:** `{gcn_loaded}`")
|
79 |
-
st.markdown(f"π **Threshold Value:** `{best_threshold}`")
|
80 |
|
81 |
# ------------------- Load Models with Spinner -------------------
|
82 |
# ------------------- Load Models with Temporary Messages -------------------
|
@@ -108,6 +106,9 @@ except Exception as e:
|
|
108 |
best_threshold = 0.5
|
109 |
st.warning(f"β οΈ Using default threshold (0.5) for GCN model. Reason: {e}")
|
110 |
|
|
|
|
|
|
|
111 |
|
112 |
|
113 |
|
|
|
74 |
# ------------------- UI Setup -------------------
|
75 |
st.set_page_config(layout="wide", page_title="Drug Toxicity Predictor")
|
76 |
st.title("π§ͺ Drug Toxicity Prediction Dashboard")
|
77 |
+
|
|
|
|
|
78 |
|
79 |
# ------------------- Load Models with Spinner -------------------
|
80 |
# ------------------- Load Models with Temporary Messages -------------------
|
|
|
106 |
best_threshold = 0.5
|
107 |
st.warning(f"β οΈ Using default threshold (0.5) for GCN model. Reason: {e}")
|
108 |
|
109 |
+
st.markdown(f"π **Fingerprint Model Loaded:** `{fp_loaded}`")
|
110 |
+
st.markdown(f"π **GCN Model Loaded:** `{gcn_loaded}`")
|
111 |
+
st.markdown(f"π **Threshold Value:** `{best_threshold}`")
|
112 |
|
113 |
|
114 |
|