diff --git a/Untitled.ipynb b/Untitled.ipynb
index 1666d3e..ace16fd 100644
--- a/Untitled.ipynb
+++ b/Untitled.ipynb
@@ -3,7 +3,7 @@
   {
    "cell_type": "code",
    "execution_count": 3,
-   "id": "730478c4",
+   "id": "34f17c80",
    "metadata": {},
    "outputs": [
     {
@@ -40,17 +40,18 @@
   {
    "cell_type": "code",
    "execution_count": 4,
-   "id": "83ca8d62",
+   "id": "8335c665",
    "metadata": {},
    "outputs": [],
    "source": [
-    "model_PATH='/home/jovyan/work/src/output/models/model2000.ckpt'"
+    "model_PATH='/home/jovyan/work/src/output/models/model2000.ckpt'\n",
+    "IMAGE_SZ = 128"
    ]
   },
   {
    "cell_type": "code",
    "execution_count": 8,
-   "id": "3de3cca5",
+   "id": "2709c7f3",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -63,7 +64,7 @@
   {
    "cell_type": "code",
    "execution_count": 11,
-   "id": "c1839d25",
+   "id": "a9d338ee",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -77,7 +78,7 @@
   {
    "cell_type": "code",
    "execution_count": 6,
-   "id": "a9b8d2ba",
+   "id": "254b5ac4",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -98,7 +99,7 @@
   {
    "cell_type": "code",
    "execution_count": 12,
-   "id": "096806cc",
+   "id": "0ad73ad4",
    "metadata": {},
    "outputs": [],
    "source": [
diff --git a/handler.py b/handler.py
index 3548bf8..2311ca4 100644
--- a/handler.py
+++ b/handler.py
@@ -9,6 +9,7 @@
 import sys
 
 model_PATH='/home/jovyan/work/src/output/models/model2000.ckpt'
+IMAGE_SZ = 128
 
 def load_demo_image(in_PATH):
     img = np.array(Image.open(in_PATH).convert('RGB'))[np.newaxis] / 255.0