Update app.py

app.py

@@ -1,7 +1,27 @@
 import gradio as gr
+import tensorflow as tf
+import numpy as input
 
-def greet(name):
-    return "Hello " + name + "!!"
+num_classes = 200
+IMG_HEIGHT = 256
+IMG_WIDTH = 256
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def normalize_image(img):
+    img = tf.cast(img, tf.float32)/255.
+    img = tf.image.resize(img, (IMG_HEIGHT, IMG_WIDTH), method='bilinear')
+    return img
+
+
+def predict_fn(img):
+    img = img.convert('RGB')
+    img_data = normalize_image(img)
+    x = np.array(img_data)
+    x = np.expand_dims(x, axis=0)
+    temp = model.predict(x)
+    return temp
+
+
+model = tf.keras.models.load_model("model.h5")
+
+interface = gr.Interface(predict_fn, gr.inputs.Image(type='PIL'), outputs='label', examples=path,)
+interface.launch()