Update app.py

app.py

@@ -21,12 +21,8 @@ def predict_fn(img):
     x = np.array(img_data)
     x = np.expand_dims(x, axis=0)
     temp = model.predict(x)
-
-    idx = np.argsort(np.squeeze(temp))[::-1]
-    top3_value = np.asarray([temp[0][i]] for i in idx[0:3])
-    top3_idx = idx[0:3]
     
-    return {CLASS_LABEL[i]:str(v) for i, v in zip(top3_idx,top3_value)}
+    return CLASS_LABEL[temp]
 
 
 model = tf.keras.models.load_model("model.h5")