Update main.py

iridia-ulb · Mar 24, 2024 · 8d14b0b · 8d14b0b
1 parent 001a677
commit 8d14b0b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Cat_or_Dog/main.py b/Cat_or_Dog/main.py
@@ -137,11 +137,11 @@ def updatePixmap(self, path, pred=1000):
         self.predLabel.setText(str(self.predictions[self.imgIndex]))
         if pred < 0.5:
             self.predLabel.setText(
-                f"I think it's a Cat! Confidence: {(0.5-pred)*200:.0f}%"
+                f"I think it's a Cat! Confidence: {(1.0-pred)*100:.0f}%"
             )
         elif pred > 0.5 and pred != 1000:
             self.predLabel.setText(
-                f"I think it's a Dog! Confidence: {(-0.5+pred)*200:.0f}%"
+                f"I think it's a Dog! Confidence: {pred*100:.0f}%"
             )
         else:
             self.predLabel.setText("I don't know yet ")