Skip to content

Commit f228a5d

Browse files
committed
update custom notebook
1 parent ad13918 commit f228a5d

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

notebooks/custom_mouse_video.ipynb

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@
99
"source": [
1010
"#If th openai api key is not set already, please set it here.\n",
1111
"import os\n",
12-
"os.environ['OPENAI_API_KEY'] = 'your key'"
12+
"if 'OPENAI_API_KEY' not in os.environ: \n",
13+
" os.environ['OPENAI_API_KEY'] = 'your key'"
1314
]
1415
},
1516
{
@@ -74,6 +75,16 @@
7475
" "
7576
]
7677
},
78+
{
79+
"cell_type": "markdown",
80+
"id": "603f97b9",
81+
"metadata": {},
82+
"source": [
83+
"### Note that unlike other notebooks, we don't have keypoint_file_path here (as it's not provided)\n",
84+
"### By default, we use gpt-4o to determine which SuperAnimal models to run and it will run SuperAnimal in the first time the keypoints related queries are asked\n",
85+
"### Make sure you use a short video clips if you are not using GPUs in Linux (Mac silicon support to be added)"
86+
]
87+
},
7788
{
7889
"cell_type": "code",
7990
"execution_count": 3,
@@ -132,7 +143,6 @@
132143
"config = Config( amadeus_root / \"amadeusgpt/configs/template.yaml\")\n",
133144
"config['video_info']['video_file_path'] = video_path\n",
134145
"config['video_info']['scene_frame_number'] = scene_frame_number\n",
135-
"config['keypoint_info']['keypoint_file_path'] = None #str(amadeus_root / config['keypoint_info']['keypoint_file_path'])\n",
136146
"print (config)\n",
137147
"amadeus = AMADEUS(config)"
138148
]

0 commit comments

Comments
 (0)