Updates inference to use correct model.

model/notebooks/inference.ipynb

@@ -51,10 +51,10 @@
     "repo_id = \"ppak10/defect-classification-t5-prompt-02-epochs\"\n",
     "\n",
     "# Initialize the model\n",
-    "model = DistilBertClassificationModel(repo_id)\n",
+    "# model = DistilBertClassificationModel(repo_id)\n",
     "# model = SciBertClassificationModel(repo_id)\n",
     "# model = LlamaClassificationModel()\n",
-    "# model = T5ClassificationModel(repo_id)"
+    "model = T5ClassificationModel(repo_id)"
    ]
   },
   {