README.md file updated

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+models/* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+myvenv/

.gradio/certificate.pem

@@ -0,0 +1,31 @@
+-----BEGIN CERTIFICATE-----
+MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
+TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
+cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
+WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
+ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
+MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
+h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
+0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
+A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
+T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
+B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
+B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
+KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
+OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
+jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
+qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
+rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
+HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
+hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
+ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
+3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
+NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
+ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
+TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
+jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
+oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
+4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
+mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
+emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
+-----END CERTIFICATE-----

README.md

@@ -12,3 +12,6 @@ short_description: Deploying  an image classification project.
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+
+gradio app URL: https://huggingface.co/spaces/fayez94/cap-recognizer

app.ipynb

@@ -0,0 +1,335 @@
+{
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "Sq8K13e0F5fx"
+      },
+      "outputs": [],
+      "source": [
+        "#The line #!default_exp app is typically used in fastai's nbdev library, which is used for creating Python libraries from Jupyter notebooks.\n",
+        "\n",
+        "#!default_exp app"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "rAQBAIwXGa65"
+      },
+      "source": [
+        "#Cap recognizer"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 1,
+      "metadata": {
+        "id": "DgI4y7YPGdEL"
+      },
+      "outputs": [],
+      "source": [
+        "!pip install -Uqq fastai gradio nbdev"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [
+        {
+          "ename": "ModuleNotFoundError",
+          "evalue": "No module named 'fastai'",
+          "output_type": "error",
+          "traceback": [
+            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+            "\u001b[0;31mModuleNotFoundError\u001b[0m                       Traceback (most recent call last)",
+            "Cell \u001b[0;32mIn[3], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mimport\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;21;01mfastai\u001b[39;00m\n\u001b[1;32m      2\u001b[0m \u001b[38;5;28mprint\u001b[39m(fastai\u001b[38;5;241m.\u001b[39m__version__)\n",
+            "\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'fastai'"
+          ]
+        }
+      ],
+      "source": [
+        "import fastai\n",
+        "print(fastai.__version__)\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 4,
+      "metadata": {},
+      "outputs": [
+        {
+          "ename": "ModuleNotFoundError",
+          "evalue": "No module named 'fastai'",
+          "output_type": "error",
+          "traceback": [
+            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+            "\u001b[0;31mModuleNotFoundError\u001b[0m                       Traceback (most recent call last)",
+            "Cell \u001b[0;32mIn[4], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;21;01mfastai\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mvision\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mall\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;28;01mimport\u001b[39;00m \u001b[38;5;241m*\u001b[39m\n",
+            "\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'fastai'"
+          ]
+        }
+      ],
+      "source": [
+        "from fastai.vision.all import *"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 10,
+      "metadata": {},
+      "outputs": [
+        {
+          "ename": "ModuleNotFoundError",
+          "evalue": "No module named 'fastai'",
+          "output_type": "error",
+          "traceback": [
+            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+            "\u001b[0;31mModuleNotFoundError\u001b[0m                       Traceback (most recent call last)",
+            "Cell \u001b[0;32mIn[10], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;21;01mfastai\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mvision\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mall\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;28;01mimport\u001b[39;00m load_learner\n\u001b[1;32m      2\u001b[0m \u001b[38;5;28;01mimport\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;21;01mgradio\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;28;01mas\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;21;01mgr\u001b[39;00m\n",
+            "\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'fastai'"
+          ]
+        }
+      ],
+      "source": [
+        "from fastai.vision.all import load_learner\n",
+        "import gradio as gr"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 11,
+      "metadata": {
+        "id": "v83ndxlVHoWj"
+      },
+      "outputs": [
+        {
+          "ename": "NameError",
+          "evalue": "name 'load_learner' is not defined",
+          "output_type": "error",
+          "traceback": [
+            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+            "\u001b[0;31mNameError\u001b[0m                                 Traceback (most recent call last)",
+            "Cell \u001b[0;32mIn[11], line 2\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[38;5;66;03m#!export\u001b[39;00m\n\u001b[0;32m----> 2\u001b[0m model \u001b[38;5;241m=\u001b[39m \u001b[43mload_learner\u001b[49m(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mmodels/cap_recognizer_v0.pkl\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n",
+            "\u001b[0;31mNameError\u001b[0m: name 'load_learner' is not defined"
+          ]
+        }
+      ],
+      "source": [
+        "#!export\n",
+        "model = load_learner(\"models/cap_recognizer_v0.pkl\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "5E15Hum7IXkU"
+      },
+      "outputs": [],
+      "source": [
+        "#|export\n",
+        "\n",
+        "cap_labels = [\n",
+        "    'balaclava cap',\n",
+        "    'baseball cap',\n",
+        "    'beanie cap',\n",
+        "    'boater hat',\n",
+        "    'bowler hat',\n",
+        "    'bucket hat',\n",
+        "    'cowboy hat',\n",
+        "    'fedora cap',\n",
+        "    'flat cap',\n",
+        "    'ivy cap',\n",
+        "    'kepi cap',\n",
+        "    'newsboy cap',\n",
+        "    'pork pie hat',\n",
+        "    'rasta cap',\n",
+        "    'sun hat',\n",
+        "    'taqiyah cap',\n",
+        "    'top hat',\n",
+        "    'trucker cap',\n",
+        "    'turban cap',\n",
+        "    'visor cap'\n",
+        "]\n",
+        "\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 17,
+      "metadata": {
+        "id": "70xb8V6tJQAU"
+      },
+      "outputs": [],
+      "source": [
+        "img = PILImage.create(\"test_images/unknown_01.jpg\")\n",
+        "img.thumbnail((128,128))"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 18,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 357
+        },
+        "id": "cGpRTaZWJhKt",
+        "outputId": "08497a21-13a9-4c76-f1c0-7af5a5df3fa3"
+      },
+      "outputs": [
+        {
+          "data": {
+            "text/html": [
+              "\n",
+              "<style>\n",
+              "    /* Turns off some styling */\n",
+              "    progress {\n",
+              "        /* gets rid of default border in Firefox and Opera. */\n",
+              "        border: none;\n",
+              "        /* Needs to be in here for Safari polyfill so background images work as expected. */\n",
+              "        background-size: auto;\n",
+              "    }\n",
+              "    progress:not([value]), progress:not([value])::-webkit-progress-bar {\n",
+              "        background: repeating-linear-gradient(45deg, #7e7e7e, #7e7e7e 10px, #5c5c5c 10px, #5c5c5c 20px);\n",
+              "    }\n",
+              "    .progress-bar-interrupted, .progress-bar-interrupted::-webkit-progress-bar {\n",
+              "        background: #F44336;\n",
+              "    }\n",
+              "</style>\n"
+            ],
+            "text/plain": [
+              "<IPython.core.display.HTML object>"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        },
+        {
+          "data": {
+            "text/html": [],
+            "text/plain": [
+              "<IPython.core.display.HTML object>"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        },
+        {
+          "data": {
+            "text/plain": [
+              "{'balaclava cap': 0.0001849675754783675,\n",
+              " 'baseball cap': 6.271898018894717e-05,\n",
+              " 'beanie cap': 0.99442058801651,\n",
+              " 'boater hat': 5.3084555418081436e-08,\n",
+              " 'bowler hat': 2.4172715029635583e-07,\n",
+              " 'bucket hat': 1.0174167073273566e-05,\n",
+              " 'cowboy hat': 8.527808148528493e-08,\n",
+              " 'fedora cap': 3.8955821679564906e-08,\n",
+              " 'flat cap': 2.141284539902699e-06,\n",
+              " 'ivy cap': 2.632383711898001e-06,\n",
+              " 'kepi cap': 0.00014104695583228022,\n",
+              " 'newsboy cap': 1.075521777238464e-05,\n",
+              " 'pork pie hat': 3.110138635520343e-08,\n",
+              " 'rasta cap': 3.2799041946418583e-05,\n",
+              " 'sun hat': 1.9477438399917446e-06,\n",
+              " 'taqiyah cap': 8.358875493286178e-05,\n",
+              " 'top hat': 3.8565694637782144e-08,\n",
+              " 'trucker cap': 4.3577329051913694e-05,\n",
+              " 'turban cap': 0.004263813607394695,\n",
+              " 'visor cap': 0.0007387499208562076}"
+            ]
+          },
+          "execution_count": 18,
+          "metadata": {},
+          "output_type": "execute_result"
+        }
+      ],
+      "source": [
+        "recognize_image(img)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 19,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "ZerTaI5OPCGx",
+        "outputId": "75d2223d-f318-4e32-a3ec-ac3fc6217534"
+      },
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "Colab notebook detected. To show errors in colab notebook, set debug=True in launch()\n",
+            "* Running on public URL: https://29d0fcc78c958c8fd7.gradio.live\n",
+            "\n",
+            "This share link expires in 72 hours. For free permanent hosting and GPU upgrades, run `gradio deploy` from the terminal in the working directory to deploy to Hugging Face Spaces (https://huggingface.co/spaces)\n"
+          ]
+        },
+        {
+          "data": {
+            "text/plain": []
+          },
+          "execution_count": 19,
+          "metadata": {},
+          "output_type": "execute_result"
+        }
+      ],
+      "source": [
+        "#!export\n",
+        "\n",
+        "image = gr.Image()\n",
+        "label = gr.Label()\n",
+        "examples = [\n",
+        "    'test_images/unknown_00.jpg',\n",
+        "    'test_images/unknown_01.jpg',\n",
+        "    'test_images/unknown_02.jpg',\n",
+        "    'test_images/unknown_03.jpg']\n",
+        "\n",
+        "\n",
+        "demo = gr.Interface(fn=recognize_image, inputs=\"image\", outputs=\"label\", examples = examples)\n",
+        "demo.launch(inline = False, share = True)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "uEfA2Pjmfl-e"
+      },
+      "source": [
+        "#Notebook to python script export\n",
+        "done it manually"
+      ]
+    }
+  ],
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "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.9.6"
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}

app.py

@@ -1,7 +1,44 @@
+from fastai.vision.all import load_learner
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+cap_labels = [
+    'balaclava cap',
+    'baseball cap',
+    'beanie cap',
+    'boater hat',
+    'bowler hat',
+    'bucket hat',
+    'cowboy hat',
+    'fedora cap',
+    'flat cap',
+    'ivy cap',
+    'kepi cap',
+    'newsboy cap',
+    'pork pie hat',
+    'rasta cap',
+    'sun hat',
+    'taqiyah cap',
+    'top hat',
+    'trucker cap',
+    'turban cap',
+    'visor cap'
+]
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+model = load_learner("models/cap_recognizer_v0.pkl")
+
+def recognize_image(image):
+  pred, idx, probs = model.predict(image)
+  return dict(zip(cap_labels, map(float,probs)))
+
+
+image = gr.Image()
+label = gr.Label()
+examples = [
+    'test_images/unknown_00.jpg',
+    'test_images/unknown_01.jpg',
+    'test_images/unknown_02.jpg',
+    'test_images/unknown_03.jpg']
+
+
+demo = gr.Interface(fn=recognize_image, inputs="image", outputs="label", examples = examples)
+demo.launch(inline = False, share = True)

models/cap_recognizer_v0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e9e0e407f17e1d15373b84f28d83e7d1aaedb46cb4358575baea40da9ce8d38c
+size 87664098

requirements.txt

@@ -0,0 +1,2 @@
+fastai==2.7.18
+gradio==5.16.0