Skip to content

Commit 52ebb5b

Browse files
authored
Update classification.ipynb
1 parent 8af54e7 commit 52ebb5b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

site/en/tutorials/images/classification.ipynb

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1088,7 +1088,8 @@
10881088
"outputs": [],
10891089
"source": [
10901090
"classify_lite = interpreter.get_signature_runner('serving_default')\n",
1091-
"classify_lite"
1091+
"print(classify_lite.get_input_details())\n",
1092+
"print(classify_lite.get_output_details())"
10921093
]
10931094
},
10941095
{
@@ -1110,7 +1111,7 @@
11101111
},
11111112
"outputs": [],
11121113
"source": [
1113-
"predictions_lite = classify_lite(sequential_1_input=img_array)['outputs']\n",
1114+
"predictions_lite = classify_lite(keras_tensor_15=img_array)['output_0']\n",
11141115
"score_lite = tf.nn.softmax(predictions_lite)"
11151116
]
11161117
},

0 commit comments

Comments
 (0)