Update app.py

app.py

@@ -139,8 +139,8 @@ if text:
     ret = {x:y for x,y in ret}
     #finbert
     ret_finbert = {keys_finbert[idxs_finbert[idx]]:sim_dict_finbert[keys_finbert[idxs_finbert[idx]]] for idx in candidate_finbert}
-    candidate_finbert = sorted(candidate_finbert.items(), key= lambda x: x[1], reverse = True)
-    candidate_finbert = {x:y for x,y in candidate_finbert}
+    ret_finbert = sorted(ret_finbert.items(), key= lambda x: x[1], reverse = True)
+    ret_finbert = {x:y for x,y in ret_finbert}
     with col1:
       st.write("distilbert-cvent")
       st.json(ret)