Also import SpanMarkerModelCardData in notebook

tomaarsen · Feb 1, 2024 · 507740d · 507740d
1 parent 9cdc18a
commit 507740d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/getting_started.ipynb b/notebooks/getting_started.ipynb
@@ -651,7 +651,7 @@
     "That was all! As simple as that. If we put it all together into a single script, it looks something like this:\n",
     "```python\n",
     "from datasets import load_dataset\n",
-    "from span_marker import SpanMarkerModel, Trainer\n",
+    "from span_marker import SpanMarkerModel, SpanMarkerModelCardData, Trainer\n",
     "from transformers import TrainingArguments\n",
     "\n",
     "def main():\n",