Added probabilities properly

app.py

@@ -5,8 +5,10 @@ learn = load_learner('model.pth')
 
 def classify_image(img):
     celebrity_name,_,probs = learn.predict(img)
-    probability = probs[list(learn.dls.vocab).index(celebrity_name)]
-    return {celebrity_name: probability}
+    results = dict(zip(list(learn.dls.vocab),probs))
+    best_results = sorted(A, key=A.get, reverse=True)[:5]
+    probabilities = {key: A[key] for key in best_results}
+    return {probabilities}
 
 image = gr.inputs.Image(shape=(224,224))
 label = gr.outputs.Label()

test.ipynb

@@ -21,7 +21,27 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 24,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "30"
+      ]
+     },
+     "execution_count": 24,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "list(learn.dls.vocab).index('Emma Watson')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 60,
    "metadata": {},
    "outputs": [
     {
@@ -64,156 +84,64 @@
     {
      "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]))"
+       "{'Emma Watson': tensor(0.9943),\n",
+       " 'Kate Beckinsale': tensor(0.0028),\n",
+       " 'Natalie Portman': tensor(0.0007),\n",
+       " 'Keira Knightley': tensor(0.0005),\n",
+       " 'Jessica Alba': tensor(0.0003)}"
       ]
      },
-     "execution_count": 13,
+     "execution_count": 60,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "learn.predict('emma_watson.jpg')"
+    "results = dict(zip(list(learn.dls.vocab),learn.predict('emma_watson.jpg')[2]))\n",
+    "best_results = sorted(A, key=A.get, reverse=True)[:5]\n",
+    "probabilities = {key: A[key] for key in best_results}\n",
+    "probabilities"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 46,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\u001b[0;31mDocstring:\u001b[0m\n",
+      "max(dim=None, keepdim=False) -> Tensor or (Tensor, Tensor)\n",
+      "\n",
+      "See :func:`torch.max`\n",
+      "\u001b[0;31mType:\u001b[0m      builtin_function_or_method"
+     ]
+    }
+   ],
+   "source": [
+    "?pred.max"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 19,
    "metadata": {},
    "outputs": [
     {
      "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']"
+       "3"
       ]
      },
-     "execution_count": 15,
+     "execution_count": 19,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "with open('celebs.txt', 'r') as file:\n",
-    "    celeb_list = file.readlines()\n",
-    "\n",
-    "celeb_list = [celeb.replace('\\n', '') for celeb in celeb_list]"
+    "celeb_list.index('Emma Watson')"
    ]
   },
   {