Update app.py
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ class TomatoLeafDiseaseDetectionApp:
|
|
69 |
confidence_score = probabilities[0, predicted_class.item()].item()
|
70 |
|
71 |
predicted_class_name = self.class_names[predicted_class.item()]
|
72 |
-
return predicted_class_name, confidence_score
|
73 |
except Exception as e:
|
74 |
return f"Error: {str(e)}", 0.0
|
75 |
|
@@ -89,7 +89,7 @@ def main():
|
|
89 |
# Predict disease
|
90 |
disease_name, confidence = app.predict_disease(image)
|
91 |
st.write(f"Predicted Disease: {disease_name}")
|
92 |
-
st.write(f"Confidence Score: {confidence:.2f}")
|
93 |
|
94 |
if __name__ == "__main__":
|
95 |
main()
|
|
|
69 |
confidence_score = probabilities[0, predicted_class.item()].item()
|
70 |
|
71 |
predicted_class_name = self.class_names[predicted_class.item()]
|
72 |
+
return predicted_class_name, confidence_score*100
|
73 |
except Exception as e:
|
74 |
return f"Error: {str(e)}", 0.0
|
75 |
|
|
|
89 |
# Predict disease
|
90 |
disease_name, confidence = app.predict_disease(image)
|
91 |
st.write(f"Predicted Disease: {disease_name}")
|
92 |
+
st.write(f"Confidence Score: {confidence:.2f}%")
|
93 |
|
94 |
if __name__ == "__main__":
|
95 |
main()
|