Upload 3 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+model.keras filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,45 @@
+import streamlit as st
+from PIL import Image
+import tensorflow as tf
+import numpy as np
+from keras.preprocessing.image import img_to_array
+import os
+
+
+@st.cache_resource
+def load_model():
+    model_path = "model.keras"  # Update with the absolute file path
+    return tf.keras.models.load_model(model_path)
+
+model = load_model()
+
+def prepare_image(img):
+    img = img.resize((220, 220))
+    img_array = img_to_array(img)
+    img_array = np.expand_dims(img_array, axis=0)
+
+    prediction = model.predict(img_array)
+    predicted_class = "Smoking" if prediction > 0.5 else "Not Smoking"
+    
+    return predicted_class, prediction[0]
+
+def run():
+    st.title("Smoking or Not Smoking Detection")
+    img_file = st.file_uploader("Choose an Image", type=["jpg", "png"])
+    
+    if img_file is not None:
+        img = Image.open(img_file).resize((250, 250))
+        st.image(img, use_column_width=False)
+        
+        # Create the directory if it doesn't exist
+        upload_dir = './upload_images/'
+        os.makedirs(upload_dir, exist_ok=True)
+        
+        save_image_path = os.path.join(upload_dir, img_file.name)
+        with open(save_image_path, "wb") as f:
+            f.write(img_file.getbuffer())
+
+        predicted_class, score = prepare_image(img)
+        st.success(f"**Predicted : {predicted_class}, Score: {score}**")
+
+run()

model.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fb76f07b54d985b3898621c5c782311d50aac3849f00ca79f55c325eabca10a
+size 402347000

requirement.txt

@@ -0,0 +1,7 @@
+requests==2.29.0
+streamlit
+numpy
+Pillow
+flask
+keras
+tensorflow