Upload app.py

app.py

@@ -391,14 +391,19 @@ def main():
                     image, model, transform
                 )
                 if predicted_class is not None:
-                    # Store results in session state
-                    st.session_state.prediction_results = {
-                        'predicted_class': predicted_class,
-                        'confidence': confidence,
-                        'probabilities': probabilities,
-                        'analysis': analyze_site_quality(predicted_class, confidence)
-                    }
-                    st.success("✅ Analysis complete!")
+                    # Confidence thresholding for OOD detection
+                    if max(probabilities) <= 0.8:
+                        st.warning("⚠️ This image does not appear to be a telecom site. Please upload a valid telecom site photo.")
+                        st.session_state.prediction_results = None
+                    else:
+                        # Store results in session state
+                        st.session_state.prediction_results = {
+                            'predicted_class': predicted_class,
+                            'confidence': confidence,
+                            'probabilities': probabilities,
+                            'analysis': analyze_site_quality(predicted_class, confidence)
+                        }
+                        st.success("✅ Analysis complete!")
     
     with col2:
         if hasattr(st.session_state, 'prediction_results'):