Load Model. Gradio UI.

.gitignore

@@ -2,7 +2,8 @@
 # Edit at https://www.toptal.com/developers/gitignore?templates=python
 
 # Trent's own python environment
-./venv_soup
+.venv_soup/
+test_img/
 
 ### Python ###
 # Byte-compiled / optimized / DLL files

app.py

@@ -1,7 +1,22 @@
 import gradio as gr
+from fastai.text.all import *
+import pathlib
+from pathlib import Path
 
-def greet(name):
-    return "Hello there " + name + "!!"
+categories = ('Korean','Vietnamese')
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+def predict_input_image(img):
+   pred,idx,probs = learn_inf.predict(img)
+   return dict(zip(categories,map(float,probs)))
+
+path = Path()
+temp = pathlib.PosixPath
+pathlib.PosixPath = pathlib.WindowsPath
+
+learn_inf = load_learner(path/'korean.pkl')
+
+image = gr.inputs.Image(shape=(180,180))
+label = gr.outputs.Label(num_top_classes=2)
+
+gr.Interface(fn=predict_input_image, inputs=image, outputs=label,interpretation='default').launch(debug='True')
 iface.launch()

korean.pkl

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

requirements.txt

@@ -1 +1,2 @@
-gradio
+gradio
+fastai