diff --git a/Untitled.ipynb b/Untitled.ipynb
index ea43d77..2ee932d 100644
--- a/Untitled.ipynb
+++ b/Untitled.ipynb
@@ -3,7 +3,7 @@
   {
    "cell_type": "code",
    "execution_count": 3,
-   "id": "654793b1",
+   "id": "89b814eb",
    "metadata": {},
    "outputs": [
     {
@@ -40,7 +40,7 @@
   {
    "cell_type": "code",
    "execution_count": 4,
-   "id": "db619e40",
+   "id": "922ad6a5",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -50,7 +50,7 @@
   {
    "cell_type": "code",
    "execution_count": 9,
-   "id": "35952a0c",
+   "id": "83e4e579",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -67,7 +67,7 @@
   {
    "cell_type": "code",
    "execution_count": 8,
-   "id": "aa544da3",
+   "id": "d953c810",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -80,7 +80,7 @@
   {
    "cell_type": "code",
    "execution_count": 11,
-   "id": "481b00f6",
+   "id": "8bd255b0",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -94,7 +94,7 @@
   {
    "cell_type": "code",
    "execution_count": 6,
-   "id": "d3aa1912",
+   "id": "5d4b9209",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -116,7 +116,7 @@
   {
    "cell_type": "code",
    "execution_count": 12,
-   "id": "e572dec9",
+   "id": "73a451eb",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -133,7 +133,7 @@
     "    image = load_demo_image(base64_str)\n",
     "    res = inference(model_PATH, image)\n",
     "    image_str = image_to_path(res)\n",
-    "    return {'Output': res}\n",
+    "    return {'Output': image_str}\n",
     "    "
    ]
   }