Create app.py

app.py

@@ -0,0 +1,17 @@
+from predict import pre_process, predict
+import gradio as gr
+from PIL import Image
+
+def predict_main(img_path):
+    image = Image.fromarray(img_path)
+    image = pre_process(image)
+    prediction = predict(image)
+    return prediction
+
+demo = gr.Interface(fn=predict_main,
+             inputs=gr.Image(),
+             outputs=gr.Label(num_top_classes=3),
+             examples=[["0008_0002.JPG"]]
+            )
+if __name__ == "__main__":
+    demo.launch()