File size: 3,351 Bytes
7c9dc7f
 
 
 
 
 
 
 
 
 
 
 
9427311
7c9dc7f
 
 
 
 
 
 
 
 
9427311
7c9dc7f
 
 
 
 
 
 
 
 
 
 
9427311
7c9dc7f
 
 
 
 
 
 
 
 
9427311
7c9dc7f
 
 
 
 
 
 
 
 
 
9427311
7c9dc7f
 
 
 
 
9427311
7c9dc7f
 
9427311
 
7c9dc7f
 
 
 
9427311
7c9dc7f
 
9427311
7c9dc7f
9427311
7c9dc7f
 
 
 
9427311
7c9dc7f
 
9427311
7c9dc7f
 
 
 
7171e7e
a3de1a8
9427311
 
7c9dc7f
 
 
 
 
9427311
7c9dc7f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "77fbe5ae",
   "metadata": {},
   "source": [
    "# Film Stars Gradio App"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "37d47b60",
   "metadata": {},
   "outputs": [],
   "source": [
    "#|default_exp app"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "96a084dc",
   "metadata": {},
   "outputs": [],
   "source": [
    "#|export\n",
    "from fastai.vision.all import *\n",
    "import gradio as gr"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0dd4db80",
   "metadata": {},
   "outputs": [],
   "source": [
    "pic = 'Penelope Cruz.jpg'\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "cf73c204",
   "metadata": {},
   "outputs": [],
   "source": [
    "#|export\n",
    "learn50 = load_learner('filmstars50.pkl')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "18bfac9a",
   "metadata": {},
   "outputs": [],
   "source": [
    "#|export\n",
    "labels = learn50.dls.vocab\n",
    "def classify_image(pic):\n",
    "    pred,pred_idx,probs = learn50.predict(PILImage.create(pic))\n",
    "    #ps, idxs = probs.topk(3)\n",
    "    return  {labels[i]: float(probs[i]) for i in range(len(labels))}\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d49e1a0e",
   "metadata": {},
   "outputs": [],
   "source": [
    "classify_image(pic);"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "773dab23",
   "metadata": {},
   "outputs": [],
   "source": [
    "#|export\n",
    "image = gr.components.Image()\n",
    "examples = ['Angelina Jolie.jpg','Penelope Cruz.jpg','Jack Nicholson.jpg']\n",
    "title = \"Identify film stars similar to your image\"\n",
    "description = \"Try uploading a photograph of yourself\"\n",
    "article = \"<p style='text-align: center'><a href='https://science4performance.com/2023/01/01/eddy-goes-to-hollywood/' target='_blank'>Blog post</a></p>\"\n",
    "intfr = gr.Interface(fn=classify_image, inputs=image, outputs=gr.components.Label(num_top_classes=3), examples=examples, title=title, description=description, article=article)\n",
    "intfr.launch(inline=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "705f241c",
   "metadata": {},
   "outputs": [],
   "source": [
    "from nbdev.export import nb_export\n",
    "nb_export('FilmStarsApp.ipynb')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "cbaa52cc",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python (base)",
   "language": "python",
   "name": "base"
  },
  "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.10.8"
  },
  "toc": {
   "nav_menu": {},
   "number_sections": true,
   "sideBar": true,
   "skip_h1_title": false,
   "title_cell": "Table of Contents",
   "title_sidebar": "Contents",
   "toc_cell": false,
   "toc_position": {},
   "toc_section_display": true,
   "toc_window_display": false
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}