Update app.py

app.py

@@ -85,7 +85,7 @@ def predict(img):
             pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
             
             # OOD Detection - modify existing probabilities instead of adding new keys
-            if (max_prob > 0.95 and entropy < 0.2) or entropy > 2.0:
+            if (max_prob > 0.95 and entropy < 0.05) or entropy > 0.1:
                 # Boost the probability of the first class and add a marker
                 pred_labels_and_probs[class_names[0]] = 0.99  # Use existing class
                 # You could also just print a warning or log it