modified: main.py

.gitignore

@@ -1 +1,3 @@
-2class_98.0.h5
+2class_98.0.h5
+*.jpeg
+*.png

main.py

@@ -94,7 +94,7 @@ def home_page():
     st.title('Brain Tumor Detection')
     st.session_state.image=st.file_uploader('Upload MRI Image',accept_multiple_files=False,type=['png', 'jpg','jpeg'],key="upload",on_change=update_photo)
     if st.session_state.image != None:
-        st.image(st.session_state.image,width=300)
+        st.image(st.session_state.image,width=150)
         col,col2=st.columns([2,3])
         radio=col.radio("Model",options=('Brain Tumor Detection','Alzheimer Detection'),key='radio',on_change=update_radio)
         check=col.checkbox('Show Prediction Scores',key='check',on_change=update_check)
@@ -109,7 +109,8 @@ def home_page():
             f.write(st.session_state.image.getbuffer())
             f.close()
             
-            col2.table(pred(st.session_state.image.name,radio,selbox,check).style.highlight_max(axis=0))
+            with st.spinner('Model Running....'):
+                col2.table(pred(st.session_state.image.name,radio,selbox,check).style.highlight_max(axis=0))