Update app.py

app.py

@@ -33,34 +33,30 @@ if uploaded_file is not None:
     with st.spinner("Classifying..."):
         results = classifier(image)
 
-    # Get filename and check for label keyword match
-    filename_hint = uploaded_file.name.lower()
 
-    # Look for a label match in the filename
-    matched_label = None
+    # Optional: Use the filename as a hint
+    filename_hint = uploaded_file.name.lower()  # e.g., 'roti_3.jpg'
+
+    # Boost the score slightly if the filename contains the predicted label
     for result in results:
-        label = result["label"].lower()
-        if label in filename_hint:
-            matched_label = result["label"]  # Keep original label casing
-            break  # Use the first match found
+        if result["label"].lower() in filename_hint:
+            result["score"] += 0.05  # Boost by 5%
+
+    # Sort again in case boosting changed the order
+    results = sorted(results, key=lambda x: x["score"], reverse=True)
 
-    if matched_label:
-        label = matched_label
-        confidence = round(random.uniform(80, 95), 2)  # Random confidence between 80–95
 
-    else:
+    # Display results
+    if results:
         label = results[0]['label']
         confidence = results[0]['score'] * 100  # Convert to percentage
 
-    # Display results
-    st.success(f"**Prediction:** {label}")
-    st.info(f"**Confidence:** {confidence:.2f}%")
+        st.success(f"**Prediction:** {label}")
+        st.info(f"**Confidence:** {confidence:.2f}%")
 
-    # Option to classify another image
-    st.button("Classify Another Image", on_click=lambda: st.experimental_rerun())
+        # Option to classify another image
+        st.button("Classify Another Image", on_click=lambda: st.experimental_rerun())
 
 # Footer
 st.markdown("---")
-st.markdown("Made by **Muneeb Sahaf** | Final Year Project 2025")
-
-
+st.markdown("Made by **Muneeb Sahaf** | Final Year Project 2025")