Spaces:

banao-tech
/

omniapi

Sleeping

App Files Files Community

gauthambalraj07@gmail.com commited on Feb 4

Commit

707dfbd

1 Parent(s): d588202

push

Browse files

Files changed (3) hide show

test.ipynb +238 -0
test/imscdr_ac_in___8.png +0 -0
test/imscdr_ac_in___9.png +0 -0

test.ipynb ADDED Viewed

	@@ -0,0 +1,238 @@

+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Processing imscdr_ac_in___8.png...\n",
+      "Status Code: 500\n",
+      "Response Text: {\"detail\":\"BaseModel.predict() got an unexpected keyword argument 'source'\"}\n",
+      "Finished processing imscdr_ac_in___8.png in 43.56 seconds\n",
+      "Processing imscdr_ac_in___9.png...\n",
+      "Status Code: 500\n",
+      "Response Text: <!DOCTYPE html>\n",
+      "<html class=\"\">\n",
+      "<head>\n",
+      "    <meta charset=\"utf-8\"/>\n",
+      "    <meta\n",
+      "            name=\"viewport\"\n",
+      "            content=\"width=device-width, initial-scale=1.0, user-scalable=no\"\n",
+      "    />\n",
+      "    <meta\n",
+      "            name=\"description\"\n",
+      "            content=\"We’re on a journey to advance and democratize artificial intelligence through open source and open science.\"\n",
+      "    />\n",
+      "    <meta property=\"fb:app_id\" content=\"1321688464574422\"/>\n",
+      "    <meta name=\"twitter:card\" content=\"summary_large_image\"/>\n",
+      "    <meta name=\"twitter:site\" content=\"@huggingface\"/>\n",
+      "    <meta\n",
+      "            property=\"og:title\"\n",
+      "            content=\"Hugging Face – The AI community building the future.\"\n",
+      "    />\n",
+      "    <meta property=\"og:type\" content=\"website\"/>\n",
+      "\n",
+      "    <title>Hugging Face – The AI community building the future.</title>\n",
+      "    <style>\n",
+      "        body {\n",
+      "            margin: 0;\n",
+      "        }\n",
+      "\n",
+      "        main {\n",
+      "            background-color: white;\n",
+      "            min-height: 100vh;\n",
+      "            padding: 7rem 1rem 8rem 1rem;\n",
+      "            text-align: center;\n",
+      "            font-family: Source Sans Pro, ui-sans-serif, system-ui, -apple-system,\n",
+      "            BlinkMacSystemFont, Segoe UI, Roboto, Helvetica Neue, Arial, Noto Sans,\n",
+      "            sans-serif, Apple Color Emoji, Segoe UI Emoji, Segoe UI Symbol,\n",
+      "            Noto Color Emoji;\n",
+      "        }\n",
+      "\n",
+      "        img {\n",
+      "            width: 6rem;\n",
+      "            height: 6rem;\n",
+      "            margin: 0 auto 1rem;\n",
+      "        }\n",
+      "\n",
+      "        h1 {\n",
+      "            font-size: 3.75rem;\n",
+      "            line-height: 1;\n",
+      "            color: rgba(31, 41, 55, 1);\n",
+      "            font-weight: 700;\n",
+      "            box-sizing: border-box;\n",
+      "            margin: 0 auto;\n",
+      "        }\n",
+      "\n",
+      "        p {\n",
+      "            color: rgba(107, 114, 128, 1);\n",
+      "            font-size: 1.125rem;\n",
+      "            line-height: 1.75rem;\n",
+      "            max-width: 28rem;\n",
+      "            box-sizing: border-box;\n",
+      "            margin: 0 auto;\n",
+      "        }\n",
+      "\n",
+      "        .dark main {\n",
+      "            background-color: rgb(11, 15, 25);\n",
+      "        }\n",
+      "\n",
+      "        .dark h1 {\n",
+      "            color: rgb(209, 213, 219);\n",
+      "        }\n",
+      "\n",
+      "        .dark p {\n",
+      "            color: rgb(156, 163, 175);\n",
+      "        }\n",
+      "    </style>\n",
+      "    <script>\n",
+      "        // On page load or when changing themes, best to add inline in `head` to avoid FOUC\n",
+      "        const key = \"_tb_global_settings\";\n",
+      "        let theme = window.matchMedia(\"(prefers-color-scheme: dark)\").matches\n",
+      "            ? \"dark\"\n",
+      "            : \"light\";\n",
+      "        try {\n",
+      "            const storageTheme = JSON.parse(window.localStorage.getItem(key)).theme;\n",
+      "            if (storageTheme) {\n",
+      "                theme = storageTheme === \"dark\" ? \"dark\" : \"light\";\n",
+      "            }\n",
+      "        } catch (e) {\n",
+      "        }\n",
+      "        if (theme === \"dark\") {\n",
+      "            document.documentElement.classList.add(\"dark\");\n",
+      "        } else {\n",
+      "            document.documentElement.classList.remove(\"dark\");\n",
+      "        }\n",
+      "    </script>\n",
+      "</head>\n",
+      "\n",
+      "<body>\n",
+      "<main>\n",
+      "    <img\n",
+      "            src=\"https://huggingface.co/front/assets/huggingface_logo.svg\"\n",
+      "            alt=\"\"\n",
+      "    />\n",
+      "    <div>\n",
+      "        <h1>500</h1>\n",
+      "        <p>Sorry, there is an error on our side.</p>\n",
+      "    </div>\n",
+      "</main>\n",
+      "</body>\n",
+      "</html>\n",
+      "\n",
+      "Finished processing imscdr_ac_in___9.png in 9.98 seconds\n",
+      "{}\n"
+     ]
+    }
+   ],
+   "source": [
+    "import os\n",
+    "import requests\n",
+    "import base64\n",
+    "from PIL import Image\n",
+    "import time\n",
+    "\n",
+    "api_url = \"https://banao-tech-omniapi.hf.space/process_image\"\n",
+    "\n",
+    "def process_image_folder(input_folder, output_folder, box_threshold=0.03, iou_threshold=0.1):\n",
+    "    if not os.path.exists(output_folder):\n",
+    "        os.makedirs(output_folder)\n",
+    "\n",
+    "    results = {}\n",
+    "    image_files = [f for f in os.listdir(input_folder) if f.lower().endswith(('.jpg', '.jpeg', '.png'))]\n",
+    "\n",
+    "    for image_file in image_files:\n",
+    "        image_path = os.path.join(input_folder, image_file)\n",
+    "        print(f\"Processing {image_file}...\")\n",
+    "\n",
+    "        start_time = time.time()\n",
+    "\n",
+    "        try:\n",
+    "            with open(image_path, \"rb\") as img_file:\n",
+    "                files = {\"image_file\": img_file}\n",
+    "                data = {\n",
+    "                    \"box_threshold\": box_threshold,\n",
+    "                    \"iou_threshold\": iou_threshold,\n",
+    "                }\n",
+    "                response = requests.post(api_url, files=files, data=data)\n",
+    "                print(\"Status Code:\", response.status_code)\n",
+    "                print(\"Response Text:\", response.text)\n",
+    "\n",
+    "            if response.status_code == 200:\n",
+    "                result = response.json()\n",
+    "                results[image_file] = {\n",
+    "                    \"parsed_content_list\": result.get(\"parsed_content_list\", []),\n",
+    "                    \"label_coordinates\": result.get(\"label_coordinates\", []),\n",
+    "                }\n",
+    "                output_image_data = base64.b64decode(result[\"image\"])\n",
+    "                output_image_path = os.path.join(output_folder, f\"labeled_{image_file}\")\n",
+    "                with open(output_image_path, \"wb\") as img_file:\n",
+    "                    img_file.write(output_image_data)\n",
+    "\n",
+    "        except Exception as e:\n",
+    "            print(f\"Error processing {image_file}: {e}\")\n",
+    "\n",
+    "        end_time = time.time()\n",
+    "        print(f\"Finished processing {image_file} in {end_time - start_time:.2f} seconds\")\n",
+    "\n",
+    "    return results\n",
+    "\n",
+    "# # Example call\n",
+    "res = process_image_folder('test', 'output_folder')\n",
+    "print(res)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{}"
+      ]
+     },
+     "execution_count": 2,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "res"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "audit",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.10"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

test/imscdr_ac_in___8.png ADDED Viewed

test/imscdr_ac_in___9.png ADDED Viewed