Update app.py

app.py

@@ -33,6 +33,19 @@ def predict_disease(image_file, model, all_labels, target_language):
 
         # Predict the class
         predictions = model.predict(img_array)
+        # Validate predictions
+        confidence_threshold = 0.98  # Require at least 98% confidence
+        confidence_scores = predictions[0]
+        max_confidence = np.max(confidence_scores)
+
+        if max_confidence < confidence_threshold:
+          print(f"Prediction confidence ({max_confidence:.2f}) is too low.")
+          return f"""
+              <h3 style="color:red; text-align:center;">
+               
+             Please upload a clearer image of the plant.
+              </h3>
+               """
         predicted_class = np.argmax(predictions[0])
         
         # Get the predicted class label