app.py

import streamlit as st
from tensorflow.keras.models import load_model
from PIL import Image
import numpy as np

model=load_model("mask.h5")

def process_image(img):
    img=img.resize((128,128))
    img=np.array(img)
    img=img/255.0
    img=np.expand_dims(img,axis=0)  # Modelin boyutunu arttırıyor
    return img

st.title("Maske Tespit Modeli")
st.write("Resim sec ve model maskeli olup olmadağını tahmin etsin")

file = st.file_uploader("Bir resim sec",type=["jpeg","jpg","png"])

if file is not None:
    img=Image.open(file)
    st.image(img,caption="yuklenen resim")
    image=process_image(img)
    prediction=model.predict(image)
    predicted_class=np.argmax(prediction)

class_names={
               0: 'without_mask',
               1: 'with_mask'
            }

st.write(class_names[predicted_class])