From ce66486e10f7bf77e43d85e30d0a9af4a678703c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Manuel=20L=C3=B3pez=20Antequera?=
 <mlopezantequera@users.noreply.github.com>
Date: Fri, 4 Jul 2025 21:34:28 +0200
Subject: [PATCH] Update YOLO_NAS_Pretrained_Export.ipynb

---
 notebooks/YOLO_NAS_Pretrained_Export.ipynb | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/notebooks/YOLO_NAS_Pretrained_Export.ipynb b/notebooks/YOLO_NAS_Pretrained_Export.ipynb
index 4e0439e9e..f914ec427 100644
--- a/notebooks/YOLO_NAS_Pretrained_Export.ipynb
+++ b/notebooks/YOLO_NAS_Pretrained_Export.ipynb
@@ -69,6 +69,20 @@
         "\n",
         "files.download('yolo_nas_s.onnx')"
       ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "labels = model.get_class_names()\n",
+        "with open(\"coco-80.txt\", \"w\") as f:\n",
+        "  f.write(\"\\n\".join(labels))\n",
+        "files.download(\"coco-80.txt\")"
+      ],
+      "metadata": {
+        "id": "KEhxK_lzvlqk"
+      },
+      "execution_count": null,
+      "outputs": []
     }
   ],
   "metadata": {