ShivamKum4r commited on
Commit
566269e
Β·
verified Β·
1 Parent(s): 7b72976

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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
- st.markdown(f"πŸ” **Fingerprint Model Loaded:** `{fp_loaded}`")
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