Skip to content

Commit bd5d1fd

Browse files
committed
update load_model_and_create_your_own.ipynb
1 parent bda47ce commit bd5d1fd

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

example/load_model_and_create_your_own.ipynb

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -322,9 +322,11 @@
322322
}
323323
],
324324
"source": [
325+
"from io import BinaryIO\n",
326+
"\n",
325327
"for i, cover in enumerate(model.covers):\n",
326328
" downloaded_cover = download(cover)\n",
327-
" cover_data: NDArray[Any] = imread(downloaded_cover.path) # type: ignore\n",
329+
" cover_data: NDArray[Any] = imread(BinaryIO(downloaded_cover.read())) # type: ignore\n",
328330
" _ = plt.figure(figsize=(10, 10))\n",
329331
" plt.imshow(cover_data) # type: ignore\n",
330332
" plt.xticks([]) # type: ignore\n",

0 commit comments

Comments
 (0)