Update app.py

app.py

@@ -20,7 +20,6 @@ st.title('Malaria Parazit Tarama')
 st.write('Resim seçin ve model tahmin etsin')
 
 file = st.file_uploader('Bir resim seçin', type=['jpg', 'jpeg', 'png'])
-class_names = ['Normal', 'Parazit']
 
 if file is not None:
     img = Image.open(file)
@@ -28,5 +27,6 @@ if file is not None:
     image = process_image(img)
     prediction = model.predict(image)
     st.write(prediction[0])
-    predicted_class = np.argmax(prediction)
-    st.write(class_names[predicted_class])
+    if prediction[0]<=.5:
+        st.write('Normal')
+    else: st.write('Parazit')