modified: main.py

main.py

@@ -62,7 +62,7 @@ def pred(img,radio,selbox,check):
     if check:
         pred=pd.DataFrame({
         'class_name' : result,
-        'pred_score' : pred.flatten()
+        'pred_score' : pred.flatten()*100
         })
         return pred
     pred = np.argmax(pred, axis=1)
@@ -93,24 +93,26 @@ def home_page():
     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)
-        radio=st.radio("Model",options=('Brain Tumor Detection','Alzheimer Detection'),key='radio',on_change=update_radio)
-        check=st.checkbox('Show Prediction Scores',key='check',on_change=update_check)
+        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)
         if radio =='Brain Tumor Detection':
-            selbox=st.selectbox("choose a number of Classes",options=('44 Classes','17 Classes' ,'15 Classes','2 Classes'),index=0,key='box',on_change=update_selbox)
+            selbox=col2.selectbox("choose a number of Classes",options=('44 Classes','17 Classes' ,'15 Classes','2 Classes'),index=0,key='box',on_change=update_selbox)
         else:
-            selbox=st.radio("choose a number of Classes",options=(['4 Classes']),index=0,key='box1',on_change=update_selbox)
-        state =st.button('Get Result')
+            selbox=col2.radio("choose a number of Classes",options=(['4 Classes']),index=0,key='box1',on_change=update_selbox)
+        
+        state =col.button('Get Result')
         if state:
             f=open(st.session_state.image.name, 'wb') 
             f.write(st.session_state.image.getbuffer())
             f.close()
             
-            st.write(pred(st.session_state.image.name,radio,selbox,check))
+            col2.write(pred(st.session_state.image.name,radio,selbox,check))
 
 
 
 def About_page():
-    st.write("Soon")
+    st.error("Nothing Here yet")
 
 def main():
     spr_sidebar()