arshenoy commited on
Commit
32d59e6
·
1 Parent(s): 899de88

Deployment of cerebAI to streamlit cloud

Browse files
Files changed (2) hide show
  1. .gitattributes +1 -0
  2. cerebAI.py +1 -2
.gitattributes ADDED
@@ -0,0 +1 @@
 
 
1
+ best_model.pth filter=lfs diff=lfs merge=lfs -text
cerebAI.py CHANGED
@@ -110,9 +110,8 @@ def plot_heatmap_and_original(original_image: np.ndarray, heatmap: np.ndarray, p
110
  plt.tight_layout()
111
  return fig
112
 
113
- # ==============================================================================
114
  # -------------------- STREAMLIT FRONTEND --------------------
115
- # ==============================================================================
116
 
117
  st.set_page_config(page_title="CerebAI: Stroke Prediction Dashboard", layout="wide")
118
  st.title("CerebAI: AI-Powered Stroke Detection")
 
110
  plt.tight_layout()
111
  return fig
112
 
 
113
  # -------------------- STREAMLIT FRONTEND --------------------
114
+
115
 
116
  st.set_page_config(page_title="CerebAI: Stroke Prediction Dashboard", layout="wide")
117
  st.title("CerebAI: AI-Powered Stroke Detection")