Merge pull request #2387 from chicco-carone:fix-classification-download

PiperOrigin-RevId: 778332905

Author: copybara-github

site/en/tutorials/images/classification.ipynb

@@ -153,8 +153,8 @@
         "import pathlib\n",
         "\n",
         "dataset_url = \"https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz\"\n",
-        "data_dir = tf.keras.utils.get_file('flower_photos.tar', origin=dataset_url, extract=True)\n",
-        "data_dir = pathlib.Path(data_dir).with_suffix('')"
+        "data_dir = tf.keras.utils.get_file(\"flower_photos.tgz\", origin=dataset_url, extract=True)\n",
+        "data_dir = pathlib.Path(data_dir).parent / \"flower_photos_extracted\" / \"flower_photos\""
       ]
     },
     {
@@ -1088,7 +1088,8 @@
       "outputs": [],
       "source": [
         "classify_lite = interpreter.get_signature_runner('serving_default')\n",
-        "classify_lite"
+        "print(classify_lite.get_input_details())\n",
+        "print(classify_lite.get_output_details())"
       ]
     },
     {
@@ -1110,7 +1111,7 @@
       },
       "outputs": [],
       "source": [
-        "predictions_lite = classify_lite(sequential_1_input=img_array)['outputs']\n",
+        "predictions_lite = classify_lite(keras_tensor_15=img_array)['output_0']\n",
         "score_lite = tf.nn.softmax(predictions_lite)"
       ]
     },
@@ -1176,7 +1177,7 @@
     "colab": {
       "collapsed_sections": [],
       "name": "classification.ipynb",
-            "toc_visible": true
+      "toc_visible": true
     },
     "kernelspec": {
       "display_name": "Python 3",