fix: updates to notebook

Author: themattinthehatt

scripts/litpose_training_demo.ipynb

@@ -98,7 +98,10 @@
     "#     ERROR: pip's dependency resolver does not currently take into account all the packages that are installed.\n",
     "# This is fine and can be ignored\n",
     "\n",
-    "!pip install -e ."
+    "!pip install -e .\n",
+    "\n",
+    "# fiftyone patch for ubuntu 22.04, which colab runs as of 07/2023\n",
+    "!pip install fiftyone-db-ubuntu2204"
    ]
   },
   {
@@ -841,7 +844,6 @@
     "        labeled_mp4_file=labeled_mp4_file,\n",
     "        trainer=trainer,\n",
     "        model=model,\n",
-    "        gpu_id=cfg.training.gpu_id,\n",
     "        data_module=data_module_pred,\n",
     "        save_heatmaps=cfg.eval.get(\"predict_vids_after_training_save_heatmaps\", False),\n",
     "    )\n",
@@ -981,6 +983,7 @@
     "# Launch the FiftyOne UI\n",
     "# - Select the dataset you just built (e.g., `lightning-pose-demo`) on the top left dropdown menu.\n",
     "# - If you receive a 403 error, be sure to enable all cookies for this site in your browser\n",
+    "# - You may need to wait up to a minute for the UI to appear\n",
     "fo.launch_app()"
    ]
   },
@@ -1024,7 +1027,7 @@
     "model_names = get_model_folders_vis(model_folders)\n",
     "\n",
     "# get prediction files for each model\n",
-    "prediction_files = update_vid_metric_files_list(video=\"test_vid\", model_preds_folder=model_folders)\n",
+    "prediction_files = update_vid_metric_files_list(video=\"test_vid\", model_preds_folders=model_folders)\n",
     "\n",
     "# load data\n",
     "dframes_metrics = defaultdict(dict)\n",