test

app.py

@@ -35,9 +35,10 @@ def predict_fn(img):
      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]
+     sorted_labels = [LABELS[i] for i in idx]
+     sorted_values = [temp[0][i] for i in idx]
      
-     return {LABELS[i]:str(v) for i,v in zip(top3_idx,top3_value)}
+     return {label: str(value) for label, value in zip(sorted_labels, sorted_values)}
+
      
 gr.Interface(predict_fn, gr.Image(type='pil'), outputs='label', examples=path,).launch()