sort tags

app.py

@@ -139,14 +139,12 @@ def create_tags(image, threshold):
         indices = torch.where(probabilities > threshold)[0]
         values = probabilities[indices]
 
-    temp = []
     tag_score = dict()
     for i in range(indices.size(0)):
-        temp.append([allowed_tags[indices[i]], values[i].item()])
         tag_score[allowed_tags[indices[i]]] = values[i].item()
-    temp = [t[0] for t in temp]
-    text_no_impl = ", ".join(temp)
-    return text_no_impl, tag_score
+    sorted_tag_score = dict(sorted(tag_score.items(), key=lambda item: item[1], reverse=True))
+    text_no_impl = ", ".join(sorted_tag_score.keys())
+    return text_no_impl, sorted_tag_score
 
 with gr.Blocks() as demo:
     gr.Markdown("""