Add Application File

app.py

@@ -1,7 +1,13 @@
+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()
+model_inf=load_learner('export.pkl')
+
+categories=('Art Deco','Byzantine','Gothic','Neoclassical')
+
+def image_classifier(img):
+    pred,pred_idx,probs = model_inf.predict(img)
+    return dict(zip(categories,map(float,probs)))
+examples_arch=['ArtDeco.jpg','Byzantine.jpg','Gothic.jpg','Neoclassical.jpg']
+gr.Interface(fn=image_classifier, inputs=gr.inputs.Image(shape=(192, 192)), outputs=gr.outputs.Label(num_top_classes=4),examples=examples_arch).launch(share=True)