diff --git a/demo_apps/OctoAI_API_examples/Llama2_Gradio.ipynb b/demo_apps/OctoAI_API_examples/Llama2_Gradio.ipynb
index 9cdd3ce3d238c99328df6b769bc028f13f9e3aaa..c76d416e83cd8c4769fd1c71881832d0b0444bb2 100644
--- a/demo_apps/OctoAI_API_examples/Llama2_Gradio.ipynb
+++ b/demo_apps/OctoAI_API_examples/Llama2_Gradio.ipynb
@@ -89,8 +89,8 @@
     "        history_langchain_format.append(HumanMessage(content=human))\n",
     "        history_langchain_format.append(AIMessage(content=ai))\n",
     "    history_langchain_format.append(HumanMessage(content=message))\n",
-    "    gpt_response = llm(message) #history_langchain_format)\n",
-    "    return gpt_response#.content\n",
+    "    llm_response = llm(message, history_langchain_format)\n",
+    "    return llm_response.content\n",
     "\n",
     "gr.ChatInterface(predict).launch()"
    ]