Update model.py

model.py

@@ -16,17 +16,29 @@ def classify(img):
         return "Provide the medical Imaging of the mentioned categories"
         st.write('________________________________________________')
     if a==1:
+        st.write('________________________________________________')
         c = bone_net(im)
+        st.write('________________________________________________')
     if a==2:
+        st.write('________________________________________________')
         c = brain_net(im)
+        st.write('________________________________________________')
     if a==3:
+        st.write('________________________________________________')
         c = Eye_net(im)
+        st.write('________________________________________________')
     if a==4:
+        st.write('________________________________________________')
         c = kidney_net(im)
+        st.write('________________________________________________')
     if a==5:
+        st.write('________________________________________________')
         c = chest_net(im)
+        st.write('________________________________________________')
     if a==6:
+        st.write('________________________________________________')
         c = skin_net(im)
+        st.write('________________________________________________')
     return c
 
 
@@ -45,7 +57,16 @@ def brain_net(img):
     model = tf.keras.models.load_model("brain.h5",compile=False)
     result = model.predict(np.array([img]))
     ans = np.argmax(result)
-    return lt[ans]
+    b = ""
+    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:
+        b="Meningioma: Most are low-grade.\n\nAvoid activities that could cause head injury, such as contact sports.\nWork with a healthcare provider to manage any underlying medical conditions, such as high blood pressure, that could impact treatment.\nBe aware of potential symptoms, such as headaches or changes in vision, and seek medical attention promptly if they occur.\nTake steps to reduce stress, which can exacerbate symptoms and affect overall health.\nFollow a healthy lifestyle, including a balanced diet, regular exercise, and adequate sleep."
+    if ans==1:
+        b = "No tumor: N/A \n\n Enjoy the Life "
+    if ans==0:
+        b = "Pituitary: Can be either benign or malignant.\n\n Work with a healthcare provider to manage any underlying medical conditions, such as diabetes, that could impact treatment.\nBe aware of potential symptoms, such as headaches or changes in vision, and seek medical attention promptly if they occur.\nFollow a healthy lifestyle, including a balanced diet, regular exercise, and adequate sleep.\nTake steps to manage any hormonal imbalances that may result from the tumor.\nAvoid exposure to radiation, as this can increase the risk of developing pituitary tumors"
+    return b
 
 def chest_net(img):
     lt = ['PNEUMONIA', 'NORMAL']