Update app.py

app.py

@@ -16,7 +16,10 @@ def yolo_inference(input_type, image, video, model_id, conf_threshold, iou_thres
     Accepts detect/seg/pose/obb/cls checkpoints and renders r.plot().
     """
     model = YOLO(model_id)
-    del model.model.model[-1].one2one_cv2
+    if getattr(model, "task", None) != "classify":
+        head = model.model.model[-1]
+        if hasattr(head, "one2one_cv2"):
+            delattr(head, "one2one_cv2")
     if input_type == "Image":
         if image is None:
             w, h = 640, 480