Update app.py

app.py

@@ -33,6 +33,19 @@ if uploaded_file is not None:
     with st.spinner("Classifying..."):
         results = classifier(image)
 
+
+    # 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:
+        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)
+
+
     # Display results
     if results:
         label = results[0]['label']