Upload 2 files

app.py

@@ -24,7 +24,7 @@ with st.form("my_form"):
             image = Image.open(uploaded_file)
             st.image(image, caption="Uploaded Image", use_column_width=True)
             st.markdown(
-                f"<h2 style='text-align: center;'>Predicted Sport: {label}</h2>",
+                f"<h2 style='text-align: center;'>{label}</h2>",
                 unsafe_allow_html=True,
             )
         else:
@@ -41,6 +41,6 @@ if sample_img_choice:
     st.image(image, caption="Image", use_column_width=True)
     label = predict_label(image, model)
     st.markdown(
-        f"<h2 style='text-align: center;'>Predicted Sport: {label}</h2>",
+        f"<h2 style='text-align: center;'>{label}</h2>",
         unsafe_allow_html=True,
     )

utils.py

@@ -109,6 +109,9 @@ def predict_label(img, model):
     resized_img = resize(img, (150, 150)).numpy().astype(int)
     exp_img = np.expand_dims(resized_img, 0)
     y_prob = model.predict(exp_img)
-    y_classes = y_prob.argmax(axis=-1)
-    label = classes[y_classes[0]]
-    return label.capitalize()
+    if y_prob.max(axis=-1) < 0.5:
+        return "Cannot predict. Please provide appropriate image."
+    else:
+        y_classes = y_prob.argmax(axis=-1)
+        label = classes[y_classes[0]]
+        return "Predicted Sport: " + label.capitalize()