rice_detection / app.py
gundogdu's picture
Upload 3 files
b5ae069 verified
raw
history blame contribute delete
802 Bytes
import streamlit as st
from tensorflow.keras.models import load_model
from PIL import Image
import numpy as np
model=load_model("my_cnn_model.h5") #modeli yükledik.
def process_image(img):
img=img.resize((68,68))
img=np.array(img)
img=img/255.0 #normalize ettik
img=np.expand_dims(img,axis=0)
return img
st.title("Rice Class Classification")
st.write("Upload a Picture")
file=st.file_uploader("Select an Image",type=["jpg","jpeg","png"])
if file is not None:
img=Image.open(file)
st.image(img,caption="Uploaded Image", use_column_width=True)
image=process_image(img)
prediction=model.predict(image)
predicted_class=np.argmax(prediction)
class_names=["Tahmin Değil","Tahmin"]
st.write(class_names[predicted_class])