spuun commited on
Commit
38696b5
1 Parent(s): 7c1e4c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def launch(img):
68
  case "r18":
69
  weight += 2
70
 
71
- print(list(zip(tm_cfg["labels"], map(lambda x: x.item(), tm_items[0]))),tf_output)
72
  return weight > 0
73
 
74
  app = gr.Interface(fn=launch, inputs="pil", outputs="text")
 
68
  case "r18":
69
  weight += 2
70
 
71
+ print(sorted(list(zip(tm_cfg["labels"], map(lambda x: x.item(), tm_items[0]))), key=lambda x: x[1]), tf_output)
72
  return weight > 0
73
 
74
  app = gr.Interface(fn=launch, inputs="pil", outputs="text")