Upload CWE_Predictor.ipynb with huggingface_hub

CWE_Predictor.ipynb

@@ -620,7 +620,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 32,
    "id": "4b087fae",
    "metadata": {},
    "outputs": [
@@ -630,8 +630,8 @@
        "\n",
        "    <div>\n",
        "      \n",
-       "      <progress value='4453' max='6978' style='width:300px; height:20px; vertical-align: middle;'></progress>\n",
-       "      [4453/6978 50:27 < 28:37, 1.47 it/s, Epoch 0.64/1]\n",
+       "      <progress value='6978' max='6978' style='width:300px; height:20px; vertical-align: middle;'></progress>\n",
+       "      [6978/6978 1:18:31, Epoch 1/1]\n",
        "    </div>\n",
        "    <table border=\"1\" class=\"dataframe\">\n",
        "  <thead>\n",
@@ -672,6 +672,20 @@
        "      <td>0.720193</td>\n",
        "      <td>0.242404</td>\n",
        "    </tr>\n",
+       "    <tr>\n",
+       "      <td>5000</td>\n",
+       "      <td>1.110300</td>\n",
+       "      <td>1.111053</td>\n",
+       "      <td>0.722934</td>\n",
+       "      <td>0.244389</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <td>6000</td>\n",
+       "      <td>1.134700</td>\n",
+       "      <td>1.082806</td>\n",
+       "      <td>0.727207</td>\n",
+       "      <td>0.251264</td>\n",
+       "    </tr>\n",
        "  </tbody>\n",
        "</table><p>"
       ],
@@ -681,6 +695,16 @@
      },
      "metadata": {},
      "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/plain": [
+       "TrainOutput(global_step=6978, training_loss=1.1011348515535433, metrics={'train_runtime': 4712.2885, 'train_samples_per_second': 23.691, 'train_steps_per_second': 1.481, 'total_flos': 2912105519756448.0, 'train_loss': 1.1011348515535433, 'epoch': 1.0})"
+      ]
+     },
+     "execution_count": 32,
+     "metadata": {},
+     "output_type": "execute_result"
     }
    ],
    "source": [
@@ -689,7 +713,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 30,
+   "execution_count": 33,
    "id": "48faf17c",
    "metadata": {},
    "outputs": [],
@@ -705,37 +729,31 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 31,
+   "execution_count": 34,
    "id": "fcb11390",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "80b4e05d305a419796600df308bf989c",
+       "model_id": "a33b852d1c594a69974bb9c3d30c014a",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "model.safetensors:   0%|          | 0.00/269M [00:00<?, ?B/s]"
+       "README.md: 0.00B [00:00, ?B/s]"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "63312fcfde9d426f94ff7f17afd68049",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "README.md: 0.00B [00:00, ?B/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "No files have been modified since last commit. Skipping to prevent empty commit.\n",
+      "No files have been modified since last commit. Skipping to prevent empty commit.\n"
+     ]
     },
     {
      "name": "stdout",
@@ -759,19 +777,10 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 35,
    "id": "33847880",
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "CWE-119\n",
-      "CWE-89\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "# Load the model\n",
     "model = AutoModelForSequenceClassification.from_pretrained(\"artifacts/model\")\n",
@@ -790,10 +799,26 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 36,
    "id": "cdeaadbb",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "CWE-119\n",
+      "CWE-89\n",
+      "CWE-79\n",
+      "CWE-287\n",
+      "CWE-22\n",
+      "CWE-190\n",
+      "CWE-401\n",
+      "CWE-77\n",
+      "CWE-326\n"
+     ]
+    }
+   ],
    "source": [
     "\n",
     "print(predict_cwe(\"Buffer overflow in POP servers allows remote attackers to gain root access using a long PASS command.\"))\n",