Add model

.vscode/launch.json

@@ -0,0 +1,16 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python: Current File",
+            "type": "python",
+            "request": "launch",
+            "program": "${file}",
+            "console": "integratedTerminal",
+            "justMyCode": true
+        }
+    ]
+}

app.py

@@ -1,7 +1,18 @@
+from fastai.vision.all import *
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+learn = load_learner('model.pkl')
+
+categories = ('Anime', 'Real')
+
+def classify_image(img):
+  pred,idx,probs = learn.predict(img)
+  return dict(zip(categories, map(float,probs)))
+
+inputs = gr.inputs.Image(shape=(192,192))
+outputs = gr.outputs.Label()
+examples = ['/content/anime.jpeg','/content/real.jpg','/content/dunno.jpg']
+
+intf = gr.Interface(fn=classify_image, inputs="image", outputs="label",examples=examples)
+intf.launch(inline=False,share=True)

model.pkl

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