Update app.py

app.py

@@ -27,14 +27,15 @@ query = st_keyup(
 )
 
 if query != "" and is_hiragana_or_katakana(query):
-    st.info("OK")
-    # if query in query_candidates:
-    #     top_candidates, top_losses = query_candidates[query]
-    # else:
-    #     top_candidates, top_losses = search_candidates(query, top_k=100)
-    # answers = ["{}: {:.2f}".format(top_candidates[index], top_losses[index]) for index in range(min(len(top_candidates), 10))]
-    # value = "<br>".join(answers)
-    # value += f"<br>({len(top_candidates)}候補)"
-    # html.value = value
+    if query in query_candidates:
+        top_candidates, top_losses = query_candidates[query]
+    else:
+        # top_candidates = [query]
+        # top_losses = [0.0]
+        top_candidates, top_losses = search_candidates(query, query_candidates, model=model, tokenizer=tokenizer, top_k=100)
+    answers = ["{}: {:.2f}".format(top_candidates[index], top_losses[index]) for index in range(min(len(top_candidates), 10))]
+    value = "\n".join(answers)
+    value += f"\n({len(top_candidates)}候補)"
+    st.info(value)
 else:
     st.info("ひらがな/カタカナのみを入力してください")