Update model

requirements.txt

@@ -0,0 +1,2 @@
+fastai
+gradio

app.py

@@ -1,7 +1,18 @@
+from fastai.vision.all import *
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+examples = ['/images/cloud1.jpg', '/images/cloud2.jpg', '/images/cloud3.jpg', '/images/cloud4.jpg', '/images/cloud5.jpg', '/images/cloud6.jpg', '/images/cloud7.jpg', '/images/cloud8.jpg', '/images/cloud9.jpg', '/images/cloud10.jpg']
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+categories = ('Cirrus', 'Stratus')
+
+learn = load_learner('model.pkl')
+
+def classify_image(img):
+    pred,idx,probs = learn.predict(img)
+    return dict(zip(categories, map(float,probs)))
+
+image = gr.Image()
+label = gr.Label()
+
+demo = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+demo.launch()

model.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9d1784e4560be0f46847f89a5371747ea20947ce3c9b22bef59038acedea39b9
-size 46967686
+oid sha256:a3f422bf1339a4f8022fc797a8ed25d6935da52e583cd30524daf51786dc4bc9
+size 46957062