first commit

.gitattributes

@@ -2,7 +2,6 @@
 *.arrow filter=lfs diff=lfs merge=lfs -text
 *.bin filter=lfs diff=lfs merge=lfs -text
 *.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
 *.ftz filter=lfs diff=lfs merge=lfs -text
 *.gz filter=lfs diff=lfs merge=lfs -text
 *.h5 filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+flagged/

README.md

@@ -10,3 +10,4 @@ pinned: false
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+

app.py

@@ -0,0 +1,37 @@
+from pathlib import Path
+
+import gradio as gr
+import numpy as np
+from fastai.vision.all import *
+
+config = {
+    "labels": [
+        "Plantation",
+        "Grassland",
+        "Smallholder Agriculture"
+    ],
+    "size": 256,
+}
+
+learn = load_learner("model.pkl")
+
+def classify_image(input):
+    _, _, prediction = learn.predict(input)
+    print(prediction)
+    outputs = {label: float(prediction[i]) for i, label in enumerate(config["labels"])}
+    # Get argmax
+    argmax_label = config["labels"][np.argmax(prediction)]
+    return argmax_label, round(outputs[argmax_label], 3) * 100
+
+
+gr.Interface(
+    fn=classify_image, 
+    inputs=gr.inputs.Image(shape=(config["size"], config["size"])),
+    outputs=[
+        gr.outputs.Textbox(label="Output of the model"),
+        gr.outputs.Textbox(label="Probability (0 - 100)")
+    ],
+    examples=[str(x) for x in Path("./").glob("*.jpg")],
+    flagging_options=["Correct label", "Incorrect label"],
+    allow_flagging="manual",
+).launch()

model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce511c43ceeb3fb89c7ffcc58c4c13c464925978159e92762c48a02c061d829f
+size 87507001

requirements.txt

@@ -0,0 +1,20 @@
+ipython==8.4.0
+jupyter==1.0.0
+matplotlib==3.5.2
+numpy==1.22.4
+pandas==1.4.2
+pydot==1.4.2
+scikit-learn==1.1.1
+scipy==1.8.1
+tensorflow==2.8.0
+protobuf==3.20.0
+sklearn
+opencv-python
+fastai
+torch==1.11.0
+timm
+transformers
+ipykernel
+huggingface_hub[fastai]
+dill
+gradio