Fix prompt

Author: NielsRogge

SAM/Fine_tune_SAM_(segment_anything)_on_a_custom_dataset.ipynb

@@ -6,7 +6,7 @@
       "provenance": [],
       "machine_shape": "hm",
       "gpuType": "T4",
-      "authorship_tag": "ABX9TyM0BmAKAtr59P6i9zVArU6g",
+      "authorship_tag": "ABX9TyOrmsHyIjsjAparHnskGGto",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -3956,7 +3956,7 @@
         "prompt = get_bounding_box(ground_truth_mask)\n",
         "\n",
         "# prepare image + box prompt for the model\n",
-        "inputs = processor(image, input_boxes=[prompt], return_tensors=\"pt\").to(device)\n",
+        "inputs = processor(image, input_boxes=[[prompt]], return_tensors=\"pt\").to(device)\n",
         "for k,v in inputs.items():\n",
         "  print(k,v.shape)"
       ],