add class name

app.py

@@ -104,6 +104,10 @@ def show_preds_image(image_path):
     dim=(width,height)
 
     outputs = model.predict(source=img_res_toshow)
+    boxes = outputs[0].boxes.cls
+    class_list = []
+    for class_n in boxes.cpu().numpy():
+        class_list.append(outputs[0].names[class_n])
     angle_list = []
     for object in range(len(outputs[0].masks.masks)):
         #obtain BW image
@@ -126,10 +130,12 @@ def show_preds_image(image_path):
             # Find the orientation of each shape
             angle_deg = getOrientation(c, img)
             angle_list.append(angle_deg)
-    text = 'The orientation of the objects are: '
-    for angle in angle_list:
-        text += str(angle) + ' '
-    text += 'degrees'
+
+    # Create a text string for the orientation of the objects and the class of the objects
+    text = ""
+    for i in range(len(angle_list)):
+        text = text + "Object " + str(i+1) + ": " + str(angle_list[i]) + " deg, " + class_list[i] + "\n"
+        
     results = outputs[0].cpu().numpy()
     for i, det in enumerate(results.boxes.xyxy):
         cv2.rectangle(