Update app.py

app.py

@@ -31,15 +31,13 @@ def predict_image(image):
     predicted_class_index = np.argmax(probabilities)
     if predicted_class_index == 0:
         predicted_label = "Clean"
-        predicted_probability = probabilities[0][0]*100
+        predicted_probability = probabilities[0][0] * 100  # Convert to percentage
     else:
         predicted_label = "Carries"
-        predicted_probability = probabilities[0][1]*100
-    #confidence = float(np.max(probabilities))
-    predicted_probability = str(predicted_probability)
-    
+        predicted_probability = probabilities[0][1] * 100  # Convert to percentage
+
     # Return the prediction result as a dictionary
-    return {"Predicted Label": predicted_label, "evalute the teath": predicted_probability}
+    return {"Predicted Label": predicted_label, "Evaluate the teeth": f"{predicted_probability:.2f}%"}
 
 
 # Create the interface