update api endpoint test code

test_api_endpoint.ipynb

@@ -26,34 +26,36 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "https://upscale-process-results.s3.amazonaws.com/1156995e-7c5f-4f0a-aa47-0dc8922f6f69.png 1156995e-7c5f-4f0a-aa47-0dc8922f6f69.png None\n"
+      "{'image_url': 'https://jiffy-staging-upscaled-images.s3.amazonaws.com/cfd6161a-4eaa-4077-b4ed-0ad0319508e6.png', 'image_key': 'cfd6161a-4eaa-4077-b4ed-0ad0319508e6.png', 'error': None}\n",
+      "https://jiffy-staging-upscaled-images.s3.amazonaws.com/cfd6161a-4eaa-4077-b4ed-0ad0319508e6.png cfd6161a-4eaa-4077-b4ed-0ad0319508e6.png None\n"
      ]
     }
    ],
    "source": [
-    "API_URL = \"https://po409l85y2ps6yo5.us-east-1.aws.endpoints.huggingface.cloud\"\n",
+    "\n",
+    "API_URL = \"https://uyw9tcuw0ydpsqmw.us-east-1.aws.endpoints.huggingface.cloud\"\n",
     "headers = {\n",
     "\t\"Accept\" : \"application/json\",\n",
-    "\t\"Content-Type\": \"application/json\" \n",
+    "\t\"Authorization\": \"Bearer hf_XXXX\",\n",
+    " \t\"Content-Type\": \"application/json\",\n",
     "}\n",
     "\n",
     "img_dir = \"test_data/\"\n",
-    "img_names = [\"4121783.png\", \"FB_IMG_1725931665635.jpg\", \"FB_IMG_1725931665635_gray.jpg\"]\n",
-    "out_scales = [10, 3, 5.49]\n",
-    "for img_name, outscale in zip(img_names, out_scales):\n",
-    "\timage_path = img_dir + img_name\n",
+    "img_urls = [\"https://jiffy-transfers.imgix.net/2/attachments/r267odvvfmkp6c5lccj1y6f9trb0?ixlib=rb-0.3.5\",\n",
+    "            # \"https://jiffy-staging-transfers.imgix.net/2/development/attachments/zo31eau0ykhbwoddrjtlbyz6w9mp?ixlib=rb-0.3.5\", # larger than > 1.96M pixels\n",
+    "            # \"https://jiffy-staging-transfers.imgix.net/2/development/attachments/b8ecchms9rr9wk3g71kfpfprqg1v?ixlib=rb-0.3.5\" # larger than > 1.96M pixels\n",
+    "           ]\n",
+    "out_scales = [3, 3, 2.49]\n",
+    "for img_url, outscale in zip(img_urls, out_scales):\n",
     "\t# create payload\n",
-    "\twith open(image_path, \"rb\") as i:\n",
-    "\t\tb64 = base64.b64encode(i.read())\n",
-    "\t\tb64 = b64.decode(\"utf-8\")\n",
-    "\t\tpayload = {\"inputs\": {\"image\": b64, \"outscale\": outscale}}\n",
+    "\tpayload = {\"inputs\": {\"image_url\": img_url, \"outscale\": outscale}}\n",
     "\t\t\n",
     "\tresponse = requests.post(API_URL, headers=headers, json=payload)\n",
     "\toutput_payload = response.json()\t\n",