Baskar2005 commited on
Commit
8e93315
·
verified ·
1 Parent(s): 7911342

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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()