Abdallahsadek commited on
Commit
42ccb1e
·
verified ·
1 Parent(s): 0144d6d

Upload 3 files

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. app.py +32 -0
  3. brain_stroke_model.keras +3 -0
  4. requirements.txt +4 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ brain_stroke_model.keras filter=lfs diff=lfs merge=lfs -text
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import streamlit as st
3
+ import tensorflow as tf
4
+ import numpy as np
5
+ from PIL import Image
6
+
7
+ # Load the model
8
+ model = tf.keras.models.load_model("brain_stroke_model.keras")
9
+
10
+ # Set title
11
+ st.title("Brain Stroke Prediction")
12
+
13
+ # Upload image
14
+ uploaded_file = st.file_uploader("Upload a brain CT scan image", type=["jpg", "jpeg", "png"])
15
+
16
+ # Prediction function
17
+ def predict(image):
18
+ image = image.resize((150, 150))
19
+ img_array = np.array(image) / 255.0
20
+ img_array = np.expand_dims(img_array, axis=0)
21
+ pred = model.predict(img_array)[0][0]
22
+ return pred
23
+
24
+ # Predict button
25
+ if uploaded_file is not None:
26
+ image = Image.open(uploaded_file)
27
+ st.image(image, caption="Uploaded Image", use_column_width=True)
28
+ prediction = predict(image)
29
+ if prediction >= 0.5:
30
+ st.error("❗ Prediction: This person is likely to have a stroke.")
31
+ else:
32
+ st.success("✅ Prediction: This person is not likely to have a stroke.")
brain_stroke_model.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a1661a858741c455e757e522d9f0d418a3e9ccb433f1e279b8baa4ba62bc019
3
+ size 57989241
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ streamlit
2
+ tensorflow
3
+ pillow
4
+ numpy