Harun01 commited on
Commit
b68ebd5
·
verified ·
1 Parent(s): 473a909

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +11 -9
src/streamlit_app.py CHANGED
@@ -13,10 +13,9 @@ def main():
13
  try:
14
  model = load_model("src/dates_classifier_model.h5")
15
  except Exception as e:
16
- st.error("❌ Model yüklenemedi. Lütfen model dosyasının doğru yolda olduğundan emin olun.")
17
  st.stop()
18
 
19
- # ✅ 9 sınıf ismi
20
  class_names = [
21
  'Rutab',
22
  'Meneifi',
@@ -33,21 +32,24 @@ def main():
33
  if file:
34
  try:
35
  image = Image.open(io.BytesIO(file.read())).convert("RGB")
36
- st.image(image, caption="Yüklenen Resim", use_column_width=True)
37
 
38
  img = image.resize((224, 224))
39
- img = np.array(img) / 255.0
 
40
  img = np.expand_dims(img, axis=0)
41
 
42
  prediction = model.predict(img)
43
  predicted_class = np.argmax(prediction)
44
 
45
- if predicted_class < len(class_names):
46
- st.success(f"Tahmin: {class_names[predicted_class]}")
47
- else:
48
- st.warning("⚠️ Tahmin edilen sınıf, sınıf isimleriyle eşleşmiyor. Lütfen modelinizi kontrol edin.")
 
 
49
  except Exception as e:
50
- st.error(f"Hata oluştu: {str(e)}")
51
 
52
  if __name__ == "__main__":
53
  main()
 
13
  try:
14
  model = load_model("src/dates_classifier_model.h5")
15
  except Exception as e:
16
+ st.error("❌ Model yüklenemedi.")
17
  st.stop()
18
 
 
19
  class_names = [
20
  'Rutab',
21
  'Meneifi',
 
32
  if file:
33
  try:
34
  image = Image.open(io.BytesIO(file.read())).convert("RGB")
35
+ st.image(image, caption="Yüklenen Resim", use_container_width=True)
36
 
37
  img = image.resize((224, 224))
38
+ img = np.array(img)
39
+ img = img / 255.0 # Bu adımı kaldırman gerekebilir, model eğitimine göre!
40
  img = np.expand_dims(img, axis=0)
41
 
42
  prediction = model.predict(img)
43
  predicted_class = np.argmax(prediction)
44
 
45
+ st.success(f"Tahmin: {class_names[predicted_class]}")
46
+
47
+ # Debug için tahmin sonuçlarını göster
48
+ st.subheader("Tahmin Skorları (Softmax Çıkışı):")
49
+ for i, score in enumerate(prediction[0]):
50
+ st.write(f"{class_names[i]}: {score:.4f}")
51
  except Exception as e:
52
+ st.error(f"Hata: {str(e)}")
53
 
54
  if __name__ == "__main__":
55
  main()