Upload app.py

app.py

@@ -0,0 +1,34 @@
+import gradio as gr
+import os
+from huggingface_hub import from_pretrained_keras
+import tensorflow as tf
+import numpy as np
+
+model = from_pretrained_keras("rushic24/bit")
+allImages = []
+directory = 'images'
+CLASSES = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']
+# iterate over files in images directory
+for filename in os.listdir(directory):
+    f = os.path.join(directory, filename)
+    if os.path.isfile(f):
+      allImages.append(f)
+
+def flower_classifier(image):
+  print(image.shape)
+  image = tf.image.resize(image, (224, 224))
+  image = image / 255.0
+  image = tf.expand_dims(image, 0)
+  pred = np.argmax(model(image))
+  label = CLASSES[pred]
+  return label
+
+
+iface = gr.Interface(flower_classifier,
+                     title = "Image Classification using BigTransfer (BiT)",
+                     inputs = gr.inputs.Image(),
+                     outputs = gr.outputs.Label(num_top_classes=5),
+                     capture_session=True,
+                     examples = allImages,
+                     )
+iface.launch(debug=True)