Update app.py

app.py

@@ -14,19 +14,20 @@ for filename in os.listdir(directory):
       allImages.append(f)
 
 def flower_classifier(image):
-  image = tf.convert_to_tensor(image, dtype=tf.int8)
+  print(image.shape)
   image = tf.image.resize(image, (224, 224))
   image = image / 255.0
   image = tf.expand_dims(image, 0)
-  pred =  model(image)
-  labeldict = dict(zip([CLASSES[i] for i in range(len(pred))], pred))
-  return labeldict
+  pred = np.argmax(model(image))
+  label = CLASSES[pred]
+  return label
 
 
 iface = gr.Interface(flower_classifier,
-                     title = "Award-Winning Deep Learning Project",
-                     examples = allImages,
+                     title = "Image Classification using BigTransfer (BiT)",
                      inputs = gr.inputs.Image(),
                      outputs = gr.outputs.Label(num_top_classes=5),
-                     capture_session=True)
+                     capture_session=True,
+                     examples = allImages[:4],
+                     )
 iface.launch(debug=True)