Add application file

app.py

@@ -53,7 +53,9 @@ def classify_image(img):
     img = tf.convert_to_tensor(img, dtype=tf.float32)
     img = tf.expand_dims(img, axis=0)
     # pred,idx,probs = model.predict(img)
-    probs = model.predict(img)
+    probs = model.predict(img).flatten()
+    # vector = np.vectorize(np.float_)
+    # probs = vector(probs)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 image = gr.inputs.Image(shape=(256, 256))