Add app

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,24 @@
+import gradio as gr
+from fastai.vision.all import *  # noqa: ruff[F403]
+
+learner = load_learner("chemical.pkl")
+
+
+def classify(img):
+    pred, idx, probs = learner.predict(img)
+
+    return dict(zip(("chemical plant", "power plant"), map(float, probs)))
+
+
+image = gr.inputs.Image(
+    label="Upload image of chemical or power plant", type="filepath", shape=(192, 192)
+)
+label = gr.outputs.Label()
+examples = ["chemical.jpg", "power.jpg"]
+
+gr.Interface(
+    classify,
+    inputs=image,
+    outputs=label,
+    title="Chemical plant? Or Not?",
+)

chemical.pkl

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