Create app.py

app.py

@@ -0,0 +1,20 @@
+import os
+import numpy as np
+import gradio as gr
+
+import tensorflow as tf
+from tensorflow.keras.models import load_model
+
+model = load_model('model.keras')
+labels = ["bird", "cat", "deer", "dog"]
+
+def pred(input_img):
+    prediction = model.predict(np.array([input_img])/255) #/255 to normalize (rgb)
+    prediction = np.argmax(prediction)
+    prediction = labels[prediction]
+    return prediction
+
+demo = gr.Interface(pred, gr.Image(), "text")
+
+if __name__ == "__main__":
+    demo.launch()