Update model.py

model.py

@@ -11,34 +11,34 @@ def classify(img):
     result = model.predict(np.array([im]))
     a = np.argmax(result)
     c=""
-    st.write('________________________________________________')
+    
     if a==0:
-        return "Provide the medical Imaging of the mentioned categories"
         st.write('________________________________________________')
+        return "Provide the medical Imaging of the mentioned categories"
+        # st.write('________________________________________________')
     if a==1:
         st.write('________________________________________________')
         c = bone_net(im)
-        st.write('________________________________________________')
+        # st.write('________________________________________________')
     if a==2:
         st.write('________________________________________________')
         c = brain_net(im)
-        st.write('________________________________________________')
+        # st.write('________________________________________________')
     if a==3:
         st.write('________________________________________________')
         c = Eye_net(im)
-        st.write('________________________________________________')
+        # st.write('________________________________________________')
     if a==4:
         st.write('________________________________________________')
         c = kidney_net(im)
-        st.write('________________________________________________')
+        # st.write('________________________________________________')
     if a==5:
         st.write('________________________________________________')
         c = chest_net(im)
-        st.write('________________________________________________')
+        # st.write('________________________________________________')
     if a==6:
         st.write('________________________________________________')
         c = skin_net(im)
-        st.write('________________________________________________')
     return c
 
 
@@ -58,6 +58,7 @@ def brain_net(img):
     result = model.predict(np.array([img]))
     ans = np.argmax(result)
     b = ""
+    st.markdown('#### Precautions to be taken:')
     if ans==3:
         b = "Glioma: Can range from low-grade (mild) to high-grade (severe).\n\nAvoid activities that could cause head injury, such as contact sports.\nTake steps to manage seizures, which can be a common symptom of gliomas.\nAvoid exposure to radiation, as this can increase the risk of developing a glioma.\nTake steps to reduce stress, which can exacerbate symptoms and affect overall health.\nWork with a healthcare provider to manage any other underlying medical conditions that could impact treatment."
     if ans==2: