zaidasim232's picture
Upload 2 files
e78ab42 verified
Raw
History Blame
1.17 kB
import streamlit as st
from tensorflow.keras.models import load_model
from PIL import Image
import numpy as np
# Modeli yükle
model = load_model('malaria_model.keras')
def process_image(img):
# Resmi RGB formatına zorla (PNG'lerdeki şeffaflık kanalı hatasını önler)
img = img.convert('RGB')
img = img.resize((64, 64))
img = np.array(img)
img = img / 255.0
img = np.expand_dims(img, axis=0)
return img
st.title('Sıtma Resmi Sınıflandırma :microscope:')
st.write('Bir resim yükleyin, modelin sıtma olup olmadığını tahmin etmesini sağlayın!')
file = st.file_uploader('Bir resim yükle', type=['jpg', 'jpeg', 'png'])
if file is not None:
img = Image.open(file)
st.image(img, caption='Yüklenen Resim', use_column_width=True)
image = process_image(img)
prediction = model.predict(image)[0][0]
# Ekrana modelin güven oranını yazdıralım (Hata çözmeye çok yardımcı olur)
st.write(f"Modelin Sıtma Olma Olasılığı: %{prediction * 100:.2f}")
predicted_class = 1 if prediction > 0.5 else 0
class_names = ['Sıtma Yok', 'Sıtma Var']
st.write(class_names[predicted_class])