Update app.py

app.py

@@ -43,8 +43,9 @@ def save_image():
             upload_image_path = "processed_image.jpg"
             cv2.imwrite(upload_image_path, opencv_image)
 
-            st.success(f"Image saved as {upload_image_path}")
-            st.success("Processing Image...")
+            st.text("Detection class probabilities")
+            st.success(f"Processing Image as {upload_image_path}")
+
 
             # label of image:
             model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')  # local model
@@ -78,8 +79,6 @@ def save_image():
             st.text("Detected Gesture")
             st.image(detect_image_pred, caption="Detected Image", use_column_width=True)
 
-            st.text("Detection class probabilities")
-
         else:
             st.text("Detection Threshold is 60")
     st.text("Detection Gesture")