model testing

app.py

@@ -1,7 +1,17 @@
 import gradio as gr
+from fastbook import *
+from fastai.vision.widgets import *
+import timm
+import gradio as gr
+categories = ('accident' , 'non-accident')
+learn = load_learner('/content/model.pkl')
+def classify(img):
+  img = img.resize(224)
+  categories,index,probs = learn.predict(img)
+  return dict(zip(categories , map(float,probs)))
+img = gr.inputs.Image(shape=(224))
+label = gr.outputs.Label()
+example = ['/content/accident.jpg' , '/content/non-accident.jpg']
 
-def greet(name):
-    return "Hello " + name + "!!"
-
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+iface = gr.Interface(fn=classify, inputs="image", outputs="label" , examples= example)
 iface.launch()

model.pkl

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

model.pkl:Zone.Identifier

@@ -0,0 +1,3 @@
+[ZoneTransfer]
+ZoneId=3
+HostUrl=about:internet

requirements.txt

@@ -0,0 +1,3 @@
+-Uqq fastbook
+fastai
+timm