From 17c98c2397f1a3b27b4e25d6474f8c981ceee9ed Mon Sep 17 00:00:00 2001
From: Sanyam Bhutani <sanyambhutani@meta.com>
Date: Thu, 21 Nov 2024 15:40:02 -0800
Subject: [PATCH] Update 03_Image_Chat101.ipynb

---
 zero_to_hero_guide/03_Image_Chat101.ipynb | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/zero_to_hero_guide/03_Image_Chat101.ipynb b/zero_to_hero_guide/03_Image_Chat101.ipynb
index f90605a5a..93042f3fc 100644
--- a/zero_to_hero_guide/03_Image_Chat101.ipynb
+++ b/zero_to_hero_guide/03_Image_Chat101.ipynb
@@ -39,13 +39,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "id": "1d293479-9dde-4b68-94ab-d0c4c61ab08c",
    "metadata": {},
    "outputs": [],
    "source": [
     "HOST = \"localhost\"  # Replace with your host\n",
-    "PORT = 5000        # Replace with your port"
+    "CLOUD_PORT = 5001       # Replace with your cloud distro port\n",
+    "MODEL_NAME='Llama3.2-11B-Vision-Instruct'"
    ]
   },
   {
@@ -59,7 +60,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "id": "8e65aae0-3ef0-4084-8c59-273a89ac9510",
    "metadata": {},
    "outputs": [],
@@ -110,7 +111,7 @@
     "    cprint(\"User> Sending image for analysis...\", \"green\")\n",
     "    response = client.inference.chat_completion(\n",
     "        messages=[message],\n",
-    "        model=\"Llama3.2-11B-Vision-Instruct\",\n",
+    "        model_id=MODEL_NAME,\n",
     "        stream=stream,\n",
     "    )\n",
     "\n",
@@ -180,7 +181,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "base",
    "language": "python",
    "name": "python3"
   },
@@ -194,7 +195,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.15"
+   "version": "3.12.2"
   }
  },
  "nbformat": 4,