model works, tracked with git lfs

.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
+room_classifier.pk1 filter=lfs diff=lfs merge=lfs -text

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

app.py

@@ -0,0 +1,43 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../drive/MyDrive/Colab Notebooks/room classifier to app.ipynb.
+
+# %% auto 0
+__all__ = ['learner', 'image', 'label', 'examples', 'intf', 'classify_image']
+
+# %% ../drive/MyDrive/Colab Notebooks/room classifier to app.ipynb 2
+import platform
+
+import fastbook
+import fastai
+from fastai.vision.widgets import *
+from fastai.callback.preds import load_learner
+from fastai.vision.all import *
+
+fastbook.setup_book()
+
+# %% ../drive/MyDrive/Colab Notebooks/room classifier to app.ipynb 3
+if platform.system().lower() == "windows":
+    import pathlib
+    posix_path = pathlib.PosixPath
+    pathlib.PosixPath = pathlib.WindowsPath
+learner = load_learner("room_classifier.pk1")
+if platform.system().lower() == "windows":
+    pathlib.PosixPath = posix_path
+
+
+# %% ../drive/MyDrive/Colab Notebooks/room classifier to app.ipynb 14
+def classify_image(img):
+    pred, idx, probs = learner.predict(img)
+    return dict(zip(learner.dls.vocab, map(float, probs)))
+
+
+
+# %% ../drive/MyDrive/Colab Notebooks/room classifier to app.ipynb 16
+import gradio as gr
+
+image = gr.inputs.Image(shape=(192, 192))
+label = gr.outputs.Label()
+out_pl = widgets.Output()
+
+examples = ["examples/test_bathroom.jfif", "examples/test_living_room.jfif", "examples/test_building.jfif"]
+intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+intf.launch()

requirements.txt

@@ -0,0 +1,4 @@
+fastbook
+fastai
+gradio
+numpy<1.24

room_classifier.pk1

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