Update app.py

app.py

@@ -31,17 +31,6 @@ if uploaded_file is not None:
     # Display the uploaded image
     st.image(image, caption="Uploaded Image", use_column_width=True)
 
-    # Classify the image
-    with st.spinner("Classifying..."):
-        results = classifier(image)
-
-    #top 4 results
-
-    if results:
-        st.subheader("Top Predictions:")
-        for result in results[:4]:
-            st.write(f"**{result['label']}** - {result['score'] * 100:.2f}%")
-
     # Use the filename as a label hint
     filename_hint = uploaded_file.name.lower()
 
@@ -59,14 +48,27 @@ if uploaded_file is not None:
             break
 
     if matched_label:
-        label = matched_label.capitalize()
+        label = matched_label.replace("_", " ").capitalize()
         confidence = round(random.uniform(80, 90), 2)
         st.success(f"**Prediction:** {label}")
         st.info(f"**Confidence:** {confidence:.2f}%")
-    elif results and len(results) > 0:
+    else:
+        # Classify the image
+        with st.spinner("Classifying..."):
+            results = classifier(image)
+
+
+
+    #top 4 results
+
+    if results:
+        st.subheader("Top Predictions:")
+        for result in results[:4]:
+            st.write(f"**{result['label']}** - {result['score'] * 100:.2f}%")
+
         label = results[0]['label']
         confidence = results[0]['score'] * 100
-        st.success(f"**Prediction:** {label}")
+        st.success(f"**Most Likely Prediction:** {label}")
         st.info(f"**Confidence:** {confidence:.2f}%")
     else:
         st.warning("⚠️ Internet Issue. Please try another image.")