Update app.py

app.py

@@ -17,7 +17,6 @@ def predict_model1(image):
 
     # make a prediction using model 1
     prediction = model1.predict(img)
-
     return prediction
 
 # define a function to predict using model 2
@@ -28,12 +27,11 @@ def predict_model2(image):
 
     # make a prediction using model 2
     prediction = model2.predict(img)
-
     return prediction
 
 # create the Streamlit app
 def app():
-    st.title('Malaria Detector - CNN Model Comparison')
+    st.title('Malaria Parasite Presence Detector Using Not One But Two Models')
     st.write('Upload an image and see the predictions of two CNN models!')
 
     # create file uploader
@@ -54,16 +52,13 @@ def app():
         prediction2 = predict_model2(image)
 
         # display the predictions
-        st.subheader('Model 1 Prediction - Transfer Learning_VGG19')
-        st.write(prediction1)
+        st.subheader('Model 1 Prediction - Transfer Learning_VGG16')
         if np.argmax(prediction1) == 0:
             st.write("Malaria Parasite Not Present in the Blood sample")
         else:
             st.write("Malaria Parasite Present in the Blood sample")
 
-
         st.subheader('Model 2 Prediction - Built from Scratch')
-        st.write(prediction2)
         if np.round(prediction2) == 1:
             st.write("Malaria Parasite Not Present in the Blood sample")
         else: