Update app.py

app.py

@@ -1,7 +1,36 @@
 import gradio as gr
+import numpy as np # linear algebra
+import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)
+import os   # operating system
+from fastai.vision.all import *   # fastai stuff
+import gradio as gr
+
+def is_cat(x): return x[0].isupper()
+
+dls = ImageDataLoaders.from_name_func('.',
+    get_image_files(path), valid_pct=0.2, seed=42,
+    label_func=is_cat,
+    item_tfms=Resize(192))
+
+learn = vision_learner(dls, resnet18, metrics=error_rate) 
+# fine_tune() method automatically uses best practices for fine tuning 
+# a pre-trained model
+learn.fine_tune(3)
+
+
+learn.export('model.pkl')
+
+learn_inf = load_learner(path/'model.pkl')
+
+
+categories =('Dog','Cat')
+def classify_image(img):
+    pred, idx, probs = learn_inf.predict(img)
+    return dict(zip(categories, map(float, probs)))
 
-def greet(name):
-    return "Hello " + name + "!!"
+image = gr.inputs.Image(shape=(192,192))
+label = gr.outputs.Label()
+examples = ['dog.jpeg','cat.jpeg','catdog.jpeg']
+intt = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()