clouds

app.py

@@ -2,21 +2,23 @@ from fastai.vision.all import *
 import gradio as gr
 import timm
 
-# Cell
-learn = load_learner('model.pkl')
+learn = load_learner("model.pkl")
 
-# Cell
 categories = learn.dls.vocab
 
+
 def classify_image(img):
-    pred,idx,probs = learn.predict(img)
-    return dict(zip(categories, map(float,probs)))
+    pred, idx, probs = learn.predict(img)
+    return dict(zip(categories, map(float, probs)))
+
 
-# Cell
 image = gr.inputs.Image(shape=(192, 192))
 label = gr.outputs.Label()
-examples = ['bong.jpeg']
+examples = [
+    "example_photos/altocumulus.jpg",
+    "example_photos/lenticular.jpg",
+    "example_photos/cirrocumulus.jpg",
+]
 
-# Cell
 intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
-intf.launch()
+intf.launch()

model.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8422d6d41d797411e391c7fab3441207d065d3538315ca615956598a3a6f6488
-size 114793249
+oid sha256:34c77e03abb5cedec651ff4ad94a81657c6a5195f07754979b4282e7cd7caee8
+size 46961485

requirements.txt

@@ -1,5 +1,5 @@
 fastai==2.7.12
-gradio==3.34.0
+gradio==3.37.0
 timm==0.9.2
 torch==2.0.1
 torchvision==0.15.2