Update app.py

app.py

@@ -30,7 +30,7 @@ if uploaded_file is not None:
     # Display the uploaded image
     st.image(image, caption="Uploaded Image", use_column_width=True)
 
-     # Classify the image
+    # Classify the image
     with st.spinner("Classifying..."):
         results = classifier(image)
 
@@ -46,18 +46,18 @@ if uploaded_file is not None:
             matched_label = label
             break
 
-    # Use matched label if found in filename
     if matched_label:
-        label = matched_label.capitalize()  # Make it look nice
+        label = matched_label.capitalize()
         confidence = round(random.uniform(80, 90), 2)
-    else:
+        st.success(f"**Prediction:** {label}")
+        st.info(f"**Confidence:** {confidence:.2f}%")
+    elif results and len(results) > 0:
         label = results[0]['label']
-        confidence = results[0]['score'] * 100  # Convert to percentage
-
-
+        confidence = results[0]['score'] * 100
         st.success(f"**Prediction:** {label}")
         st.info(f"**Confidence:** {confidence:.2f}%")
-
+    else:
+        st.warning("⚠️ Could not generate a prediction. Please try another image.")
         # Option to classify another image
         st.button("Classify Another Image", on_click=lambda: st.experimental_rerun())