Update app.py

app.py

@@ -33,6 +33,18 @@ def predict_disease(image_file, model, all_labels, target_language):
 
         # Predict the class
         predictions = model.predict(img_array)
+        predictions = model.predict(img_array)
+        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