initial commit

app.py

@@ -0,0 +1,37 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: os_identify.ipynb.
+
+# %% auto 0
+__all__ = ['btn_upload', 'btn_run', 'path', 'learn_inf', 'out_pl', 'lbl_pred', 'on_click_classify']
+
+# %% os_identify.ipynb 4
+from fastai.vision.all import *
+from fastai.vision.widgets import *
+import gradio as gr
+
+# %% os_identify.ipynb 5
+def on_click_classify(change):
+    img = PILImage.create(btn_upload.data[-1])
+    out_pl.clear_output()
+    with out_pl: display(img.to_thumb(128,128))
+    pred,pred_idx,probs = learn_inf.predict(btn_upload.data[-1])
+    lbl_pred.value = f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'
+
+# %% os_identify.ipynb 6
+btn_upload = widgets.FileUpload()
+btn_run = widgets.Button(description='Classify')
+btn_run.on_click(on_click_classify)
+
+# %% os_identify.ipynb 7
+# load the model
+path = Path()
+learn_inf = load_learner(path/'os_model.pkl')
+
+# %% os_identify.ipynb 8
+out_pl = widgets.Output()
+out_pl.clear_output()
+lbl_pred = widgets.Label()
+
+
+# %% os_identify.ipynb 9
+VBox([widgets.Label('Select your screencap!'),
+      btn_upload, btn_run, out_pl, lbl_pred])

os_model.pkl

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

requirements.txt

@@ -0,0 +1 @@
+fastai