app.py

app.py

@@ -1,7 +1,47 @@
+#!/usr/bin/env python
+
+# In[12]:
+
 import gradio as gr
+from fastai.vision.all import *
+
+# In[13]:
+
+
+# In[14]:
+
+learn = load_learner("model.pkl")
+
+# In[15]:
+
+categories = {"Ronaldo", "messi", "Michael Jordan", "Rafael Nadal", "Roger Federer"}
+
+
+def classify_athlete(img):
+	pred, idx, probs = learn.predict(img)
+	return dict(zip(categories, map(float, probs)))
+
+
+# In[21]:
+
+classify_athlete("cr7.jpeg")
+
+# In[16]:
+
+image = gr.inputs.Image(shape=(192, 192))
+label = gr.outputs.Label()
+examples = ["cr7.jpeg", "messi.jpeg", "michael_jordan.jpeg"]
+
+intf = gr.Interface(fn=classify_athlete, inputs=image, outputs=label, examples=examples)
+intf.launch(inline=False)
+
+# # Making and exporting a python script
+
+# In[18]:
+
+# Export Notebook to python Script -- nbdev --
+from notebook2script import convert_notebook
 
-def greet(name):
-    return "Hello " + name + "!!"
+convert_notebook("app.ipynb", "app.py")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+# In[ ]: