Add probability

app.py

@@ -1,11 +1,12 @@
 from fastai.vision.all import *
 import gradio as gr
 
-learn = load_learner('model.pkl')
+learn = load_learner('model.pth')
 
 def classify_image(img):
     celebrity_name,_,probs = learn.predict(img)
-    return celebrity_name
+    probability = probs[list(learn.dls.vocab).index(celebrity_name)]
+    return {celebrity_name: probability}
 
 image = gr.inputs.Image(shape=(224,224))
 label = gr.outputs.Label()

celebs.txt

@@ -0,0 +1,100 @@
+Johnny Depp
+Arnold Schwarzenegger
+Jim Carrey
+Emma Watson
+Robert Downey Jr.
+Daniel Radcliffe
+Chris Evans
+Leonardo DiCaprio
+Tom Cruise
+Brad Pitt
+Charles Chaplin
+Morgan Freeman
+Tom Hanks
+Hugh Jackman
+Matt Damon
+Sylvester Stallone
+Will Smith
+Clint Eastwood
+Cameron Diaz
+George Clooney
+Steven Spielberg
+Harrison Ford
+Robert De Niro
+Al Pacino
+Robert Downey Jr.
+Russell Crowe
+Liam Neeson
+Kate Winslet
+Mark Wahlberg
+Natalie Portman
+Pierce Brosnan
+Sean Connery
+Orlando Bloom
+Dwayne Johnson
+Jackie Chan
+Angelina Jolie
+Adam Sandler
+Scarlett Johansson
+Heath Ledger
+Anne Hathaway
+Jessica Alba
+Edward Norton
+Keira Knightley
+Bradley Cooper
+Will Ferrell
+Julia Roberts
+Nicolas Cage
+Daniel Craig
+Keanu Reeves
+Ian McKellen
+Halle Berry
+Bruce Willis
+Samuel L. Jackson
+Ben Stiller
+Tommy Lee Jones
+Antonio Banderas
+Denzel Washington
+Steve Carell
+Shia LaBeouf
+Megan Fox
+James Franco
+Mel Gibson
+Vin Diesel
+Tim Allen
+Robin Williams
+Kevin Spacey
+Jason Biggs
+Seann William Scott
+Jean-Claude Van Damme
+Zach Galifianakis
+Owen Wilson
+Christian Bale
+Peter Jackson
+Sandra Bullock
+Bruce Lee
+Drew Barrymore
+Macaulay Culkin
+Jack Nicholson
+Bill Murray
+Sigourney Weaver
+Jake Gyllenhaal
+Jason Statham
+Jet Li
+Kate Beckinsale
+Rowan Atkinson
+Marlon Brando
+John Travolta
+Channing Tatum
+Ben Affleck
+Shah Rukh Khan
+Jennifer Aniston
+Emma Stone
+Chris Hemsworth
+James McAvoy
+James Cameron
+Amitabh Bachchan
+Brendan Fraser
+Rachel McAdams
+Tom Hiddleston
+Aamir Khan

test.ipynb

@@ -2,35 +2,218 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [],
    "source": [
-    "from fastai.vision.all import *"
+    "from fastai.vision.all import *\n",
+    "from pathlib import Path"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 12,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "learn = load_learner('model.pkl')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "metadata": {},
+   "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": [
+       "('Emma Watson',\n",
+       " tensor(30),\n",
+       " tensor([1.1929e-07, 5.8495e-08, 3.8198e-07, 9.0753e-06, 2.2058e-04, 1.2250e-04,\n",
+       "         2.0016e-07, 4.8305e-07, 2.2462e-08, 5.0742e-07, 3.7518e-07, 5.0198e-06,\n",
+       "         3.4834e-07, 2.3815e-06, 3.7357e-06, 1.9106e-08, 1.0254e-05, 1.1662e-04,\n",
+       "         8.8360e-07, 6.3956e-06, 1.2606e-06, 1.0974e-07, 6.6237e-08, 4.8988e-06,\n",
+       "         3.1586e-06, 3.8812e-08, 3.1651e-05, 2.0162e-06, 4.4486e-07, 1.3981e-04,\n",
+       "         9.9433e-01, 9.9587e-08, 4.3374e-06, 7.6844e-08, 1.2182e-06, 9.4450e-07,\n",
+       "         7.8193e-09, 1.0388e-06, 4.0273e-07, 1.4351e-07, 3.5202e-07, 1.0416e-07,\n",
+       "         5.1168e-07, 1.0324e-06, 8.8179e-07, 1.0323e-07, 1.7276e-05, 2.7315e-04,\n",
+       "         3.0939e-07, 2.0662e-06, 8.7540e-08, 2.7940e-06, 7.7654e-05, 2.8455e-03,\n",
+       "         1.5032e-04, 2.5726e-07, 4.8556e-04, 1.2773e-06, 2.7566e-06, 1.0871e-06,\n",
+       "         2.8510e-06, 1.8711e-07, 8.1365e-07, 1.4328e-07, 6.6794e-06, 5.6703e-08,\n",
+       "         9.1386e-09, 6.7899e-04, 2.3621e-07, 2.4263e-05, 1.7181e-07, 2.1496e-06,\n",
+       "         6.3289e-08, 1.0691e-04, 5.2058e-07, 1.4683e-06, 3.1002e-09, 1.3486e-06,\n",
+       "         6.8411e-07, 9.8865e-07, 1.4920e-04, 1.5001e-05, 1.8135e-07, 4.3004e-05,\n",
+       "         2.2933e-07, 1.4821e-06, 1.7363e-05, 5.0178e-07, 2.2853e-07, 4.7195e-08,\n",
+       "         2.7670e-07, 4.9980e-05, 4.5713e-08, 2.0050e-06, 1.3772e-06, 4.8403e-07,\n",
+       "         1.6381e-08, 3.9686e-08, 4.5362e-07]))"
+      ]
+     },
+     "execution_count": 13,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "learn.predict('emma_watson.jpg')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 15,
    "metadata": {},
    "outputs": [
     {
-     "ename": "UnpicklingError",
-     "evalue": "invalid load key, 'v'.",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mUnpicklingError\u001b[0m                           Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[2], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m learner \u001b[39m=\u001b[39m load_learner(\u001b[39m'\u001b[39;49m\u001b[39mmodel.pth\u001b[39;49m\u001b[39m'\u001b[39;49m)\n",
-      "File \u001b[0;32m~/ENTER/envs/fastai-env/lib/python3.11/site-packages/fastai/learner.py:446\u001b[0m, in \u001b[0;36mload_learner\u001b[0;34m(fname, cpu, pickle_module)\u001b[0m\n\u001b[1;32m    444\u001b[0m distrib_barrier()\n\u001b[1;32m    445\u001b[0m map_loc \u001b[39m=\u001b[39m \u001b[39m'\u001b[39m\u001b[39mcpu\u001b[39m\u001b[39m'\u001b[39m \u001b[39mif\u001b[39;00m cpu \u001b[39melse\u001b[39;00m default_device()\n\u001b[0;32m--> 446\u001b[0m \u001b[39mtry\u001b[39;00m: res \u001b[39m=\u001b[39m torch\u001b[39m.\u001b[39;49mload(fname, map_location\u001b[39m=\u001b[39;49mmap_loc, pickle_module\u001b[39m=\u001b[39;49mpickle_module)\n\u001b[1;32m    447\u001b[0m \u001b[39mexcept\u001b[39;00m \u001b[39mAttributeError\u001b[39;00m \u001b[39mas\u001b[39;00m e: \n\u001b[1;32m    448\u001b[0m     e\u001b[39m.\u001b[39margs \u001b[39m=\u001b[39m [\u001b[39mf\u001b[39m\u001b[39m\"\u001b[39m\u001b[39mCustom classes or functions exported with your `Learner` not available in namespace.\u001b[39m\u001b[39m\\\u001b[39m\u001b[39mRe-declare/import before loading:\u001b[39m\u001b[39m\\n\u001b[39;00m\u001b[39m\\t\u001b[39;00m\u001b[39m{\u001b[39;00me\u001b[39m.\u001b[39margs[\u001b[39m0\u001b[39m]\u001b[39m}\u001b[39;00m\u001b[39m\"\u001b[39m]\n",
-      "File \u001b[0;32m~/ENTER/envs/fastai-env/lib/python3.11/site-packages/torch/serialization.py:815\u001b[0m, in \u001b[0;36mload\u001b[0;34m(f, map_location, pickle_module, weights_only, **pickle_load_args)\u001b[0m\n\u001b[1;32m    813\u001b[0m     \u001b[39mexcept\u001b[39;00m \u001b[39mRuntimeError\u001b[39;00m \u001b[39mas\u001b[39;00m e:\n\u001b[1;32m    814\u001b[0m         \u001b[39mraise\u001b[39;00m pickle\u001b[39m.\u001b[39mUnpicklingError(UNSAFE_MESSAGE \u001b[39m+\u001b[39m \u001b[39mstr\u001b[39m(e)) \u001b[39mfrom\u001b[39;00m \u001b[39mNone\u001b[39;00m\n\u001b[0;32m--> 815\u001b[0m \u001b[39mreturn\u001b[39;00m _legacy_load(opened_file, map_location, pickle_module, \u001b[39m*\u001b[39;49m\u001b[39m*\u001b[39;49mpickle_load_args)\n",
-      "File \u001b[0;32m~/ENTER/envs/fastai-env/lib/python3.11/site-packages/torch/serialization.py:1033\u001b[0m, in \u001b[0;36m_legacy_load\u001b[0;34m(f, map_location, pickle_module, **pickle_load_args)\u001b[0m\n\u001b[1;32m   1027\u001b[0m \u001b[39mif\u001b[39;00m \u001b[39mnot\u001b[39;00m \u001b[39mhasattr\u001b[39m(f, \u001b[39m'\u001b[39m\u001b[39mreadinto\u001b[39m\u001b[39m'\u001b[39m) \u001b[39mand\u001b[39;00m (\u001b[39m3\u001b[39m, \u001b[39m8\u001b[39m, \u001b[39m0\u001b[39m) \u001b[39m<\u001b[39m\u001b[39m=\u001b[39m sys\u001b[39m.\u001b[39mversion_info \u001b[39m<\u001b[39m (\u001b[39m3\u001b[39m, \u001b[39m8\u001b[39m, \u001b[39m2\u001b[39m):\n\u001b[1;32m   1028\u001b[0m     \u001b[39mraise\u001b[39;00m \u001b[39mRuntimeError\u001b[39;00m(\n\u001b[1;32m   1029\u001b[0m         \u001b[39m\"\u001b[39m\u001b[39mtorch.load does not work with file-like objects that do not implement readinto on Python 3.8.0 and 3.8.1. \u001b[39m\u001b[39m\"\u001b[39m\n\u001b[1;32m   1030\u001b[0m         \u001b[39mf\u001b[39m\u001b[39m\"\u001b[39m\u001b[39mReceived object of type \u001b[39m\u001b[39m\\\"\u001b[39;00m\u001b[39m{\u001b[39;00m\u001b[39mtype\u001b[39m(f)\u001b[39m}\u001b[39;00m\u001b[39m\\\"\u001b[39;00m\u001b[39m. Please update to Python 3.8.2 or newer to restore this \u001b[39m\u001b[39m\"\u001b[39m\n\u001b[1;32m   1031\u001b[0m         \u001b[39m\"\u001b[39m\u001b[39mfunctionality.\u001b[39m\u001b[39m\"\u001b[39m)\n\u001b[0;32m-> 1033\u001b[0m magic_number \u001b[39m=\u001b[39m pickle_module\u001b[39m.\u001b[39;49mload(f, \u001b[39m*\u001b[39;49m\u001b[39m*\u001b[39;49mpickle_load_args)\n\u001b[1;32m   1034\u001b[0m \u001b[39mif\u001b[39;00m magic_number \u001b[39m!=\u001b[39m MAGIC_NUMBER:\n\u001b[1;32m   1035\u001b[0m     \u001b[39mraise\u001b[39;00m \u001b[39mRuntimeError\u001b[39;00m(\u001b[39m\"\u001b[39m\u001b[39mInvalid magic number; corrupt file?\u001b[39m\u001b[39m\"\u001b[39m)\n",
-      "\u001b[0;31mUnpicklingError\u001b[0m: invalid load key, 'v'."
-     ]
+     "data": {
+      "text/plain": [
+       "['Johnny Depp\\n',\n",
+       " 'Arnold Schwarzenegger\\n',\n",
+       " 'Jim Carrey\\n',\n",
+       " 'Emma Watson\\n',\n",
+       " 'Robert Downey Jr.\\n',\n",
+       " 'Daniel Radcliffe\\n',\n",
+       " 'Chris Evans\\n',\n",
+       " 'Leonardo DiCaprio\\n',\n",
+       " 'Tom Cruise\\n',\n",
+       " 'Brad Pitt\\n',\n",
+       " 'Charles Chaplin\\n',\n",
+       " 'Morgan Freeman\\n',\n",
+       " 'Tom Hanks\\n',\n",
+       " 'Hugh Jackman\\n',\n",
+       " 'Matt Damon\\n',\n",
+       " 'Sylvester Stallone\\n',\n",
+       " 'Will Smith\\n',\n",
+       " 'Clint Eastwood\\n',\n",
+       " 'Cameron Diaz\\n',\n",
+       " 'George Clooney\\n',\n",
+       " 'Steven Spielberg\\n',\n",
+       " 'Harrison Ford\\n',\n",
+       " 'Robert De Niro\\n',\n",
+       " 'Al Pacino\\n',\n",
+       " 'Robert Downey Jr.\\n',\n",
+       " 'Russell Crowe\\n',\n",
+       " 'Liam Neeson\\n',\n",
+       " 'Kate Winslet\\n',\n",
+       " 'Mark Wahlberg\\n',\n",
+       " 'Natalie Portman\\n',\n",
+       " 'Pierce Brosnan\\n',\n",
+       " 'Sean Connery\\n',\n",
+       " 'Orlando Bloom\\n',\n",
+       " 'Dwayne Johnson\\n',\n",
+       " 'Jackie Chan\\n',\n",
+       " 'Angelina Jolie\\n',\n",
+       " 'Adam Sandler\\n',\n",
+       " 'Scarlett Johansson\\n',\n",
+       " 'Heath Ledger\\n',\n",
+       " 'Anne Hathaway\\n',\n",
+       " 'Jessica Alba\\n',\n",
+       " 'Edward Norton\\n',\n",
+       " 'Keira Knightley\\n',\n",
+       " 'Bradley Cooper\\n',\n",
+       " 'Will Ferrell\\n',\n",
+       " 'Julia Roberts\\n',\n",
+       " 'Nicolas Cage\\n',\n",
+       " 'Daniel Craig\\n',\n",
+       " 'Keanu Reeves\\n',\n",
+       " 'Ian McKellen\\n',\n",
+       " 'Halle Berry\\n',\n",
+       " 'Bruce Willis\\n',\n",
+       " 'Samuel L. Jackson\\n',\n",
+       " 'Ben Stiller\\n',\n",
+       " 'Tommy Lee Jones\\n',\n",
+       " 'Antonio Banderas\\n',\n",
+       " 'Denzel Washington\\n',\n",
+       " 'Steve Carell\\n',\n",
+       " 'Shia LaBeouf\\n',\n",
+       " 'Megan Fox\\n',\n",
+       " 'James Franco\\n',\n",
+       " 'Mel Gibson\\n',\n",
+       " 'Vin Diesel\\n',\n",
+       " 'Tim Allen\\n',\n",
+       " 'Robin Williams\\n',\n",
+       " 'Kevin Spacey\\n',\n",
+       " 'Jason Biggs\\n',\n",
+       " 'Seann William Scott\\n',\n",
+       " 'Jean-Claude Van Damme\\n',\n",
+       " 'Zach Galifianakis\\n',\n",
+       " 'Owen Wilson\\n',\n",
+       " 'Christian Bale\\n',\n",
+       " 'Peter Jackson\\n',\n",
+       " 'Sandra Bullock\\n',\n",
+       " 'Bruce Lee\\n',\n",
+       " 'Drew Barrymore\\n',\n",
+       " 'Macaulay Culkin\\n',\n",
+       " 'Jack Nicholson\\n',\n",
+       " 'Bill Murray\\n',\n",
+       " 'Sigourney Weaver\\n',\n",
+       " 'Jake Gyllenhaal\\n',\n",
+       " 'Jason Statham\\n',\n",
+       " 'Jet Li\\n',\n",
+       " 'Kate Beckinsale\\n',\n",
+       " 'Rowan Atkinson\\n',\n",
+       " 'Marlon Brando\\n',\n",
+       " 'John Travolta\\n',\n",
+       " 'Channing Tatum\\n',\n",
+       " 'Ben Affleck\\n',\n",
+       " 'Shah Rukh Khan\\n',\n",
+       " 'Jennifer Aniston\\n',\n",
+       " 'Emma Stone\\n',\n",
+       " 'Chris Hemsworth\\n',\n",
+       " 'James McAvoy\\n',\n",
+       " 'James Cameron\\n',\n",
+       " 'Amitabh Bachchan\\n',\n",
+       " 'Brendan Fraser\\n',\n",
+       " 'Rachel McAdams\\n',\n",
+       " 'Tom Hiddleston\\n',\n",
+       " 'Aamir Khan\\n']"
+      ]
+     },
+     "execution_count": 15,
+     "metadata": {},
+     "output_type": "execute_result"
     }
    ],
    "source": [
-    "learner = load_learner('model.pkl')"
+    "with open('celebs.txt', 'r') as file:\n",
+    "    celeb_list = file.readlines()\n",
+    "\n",
+    "celeb_list = [celeb.replace('\\n', '') for celeb in celeb_list]"
    ]
   },
   {