Update app.py

app.py

@@ -8,14 +8,8 @@ import io
 from collections import Counter
 import numpy as np
 
-def load_image():
-    uploaded_file = st.file_uploader(label='Pick an image to test')
-    if uploaded_file is not None:
-        image_data = uploaded_file.getvalue()
-        st.image(image_data)
-
 def load_models():
-    model_name = 'Model/mango_model.h5'
+    model_name = 'Model/mango_new_model.h5'
     model = tf.keras.models.load_model(model_name) 
     return model
     
@@ -43,7 +37,7 @@ def predict(model, img):
     if prediction[0]>0.5:
         return 'ripe'
     else:
-        return 'ripe'
+        return 'unripe'
 
 def main():
     st.title('Mango Ripeness Classifier 🥭')