Update app.py

app.py

@@ -17,15 +17,15 @@ IMG_WIDTH, IMG_HEIGHT = 64, 48
 # Class labels
 CLASS_LABELS = {
     0: "The person in the uploaded image is driving safely",
-    1: "The person in the uploaded image is texting in the right direction and thus, distracted.",
-    2: "The person in the uploaded image is talking on the phone in the right direction and thus, distracted.",
-    3: "The person in the uploaded image is texting in the left direction and thus, distracted.",
-    4: "The person in the uploaded image is talking on the phone in the left direction and thus, distracted.",
-    5: "The person in the uploaded image is operating the radio and thus, distracted.",
-    6: "The person in the uploaded image is drinking and thus, distracted.",
-    7: "The person in the uploaded image is reaching behind and thus, distracted.",
-    8: "The person in the uploaded image is doing hair and makeup and thus, distracted.",
-    9: "The person in the uploaded image is talking to a passenger and thus, distracted."
+    1: "The person in the uploaded image is texting in the right direction and thus, distracted",
+    2: "The person in the uploaded image is talking on the phone in the right direction and thus, distracted",
+    3: "The person in the uploaded image is texting in the left direction and thus, distracted",
+    4: "The person in the uploaded image is talking on the phone in the left direction and thus, distracted",
+    5: "The person in the uploaded image is operating the radio and thus, distracted",
+    6: "The person in the uploaded image is drinking and thus, distracted",
+    7: "The person in the uploaded image is reaching behind and thus, distracted",
+    8: "The person in the uploaded image is doing hair and makeup and thus, distracted",
+    9: "The person in the uploaded image is talking to a passenger and thus, distracted"
 }
 
 def predict_image(image):
@@ -64,7 +64,7 @@ if uploaded_file is not None:
     with st.spinner("Predicting..."):
         predicted_class, confidence = predict_image(resized_image)
 
-    prediction_text = f"{predicted_class}. Confidence: {confidence:.2%}"
+    prediction_text = f"{predicted_class}. Prediction confidence: {confidence:.2%}"
     st.subheader("Prediction")
     st.write(prediction_text)