Skip to content

Commit 623e99a

Browse files
committed
update
1 parent c655c5e commit 623e99a

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

examples/enrichment-zero-shot-detection.ipynb

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -434,8 +434,8 @@
434434
"df = fd.enrich(task='zero-shot-detection', \n",
435435
" model='grounding-dino', \n",
436436
" input_df=df, \n",
437-
" input_col='ram_tags', \n",
438-
" device=\"cuda\")"
437+
" input_col='ram_tags'\n",
438+
" )"
439439
]
440440
},
441441
{
@@ -638,7 +638,8 @@
638638
" bbox_col='grounding_dino_bboxes', # column specifying bounding boxes\n",
639639
" scores_col='grounding_dino_scores', # column specifying label scores\n",
640640
" labels_col='grounding_dino_labels', # column specifying label text\n",
641-
" num_rows=10) # the number of rows in the dataframe to plot"
641+
" num_rows=10 # the number of rows in the dataframe to plot\n",
642+
") "
642643
]
643644
},
644645
{
@@ -1118,7 +1119,8 @@
11181119
" bbox_col='grounding_dino_bboxes', \n",
11191120
" scores_col='grounding_dino_scores', \n",
11201121
" labels_col='grounding_dino_labels', \n",
1121-
" num_rows=10)"
1122+
" num_rows=10\n",
1123+
")"
11221124
]
11231125
},
11241126
{
@@ -1230,7 +1232,8 @@
12301232
"results = model.run_inference(image_path=\"coco_minitrain_25k/images/val2017/000000449996.jpg\",\n",
12311233
" text_prompt=\"air field . airliner . plane . airport . airport runway . airport terminal . jet . land . park . raceway . sky . tarmac . terminal\",\n",
12321234
" box_threshold=0.3,\n",
1233-
" text_threshold=0.25)"
1235+
" text_threshold=0.25\n",
1236+
" )"
12341237
]
12351238
},
12361239
{

0 commit comments

Comments
 (0)