Update Colab_inference_demo.ipynb

AdaptiveMotorControlLab · Dec 22, 2024 · 5eeac6f · 5eeac6f
1 parent 9684ff6
commit 5eeac6f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/Colab_inference_demo.ipynb b/notebooks/Colab_inference_demo.ipynb
@@ -293,7 +293,7 @@
       "model_info = ModelInfo(\n",
         "      name=model_selection,\n",
         "      model_input_size=64 if model_selection == "SegResNet" else [64,64,64],\n",
-        "     num_classes=2,"\n",
+        "     num_classes=2,\n",
         ")"
       "inference_config.model_info = model_info"
         "result = cs3d.inference_on_images(\n",