Update app.py

app.py

@@ -20,8 +20,13 @@ sim_model.add_corpus(load_file('corpus.txt'))
 
 def ai_text(query):
     res = sim_model.most_similar(queries=query, topn=5)
-
-    res_show = str(res)
+    print(res)
+    for q_id, c in enumerate(res):
+        print('query:', query)
+        print("search top 5:")
+        print(f'\t{c}')
+    res_show = '\n'.join(
+        ['search top5：'] + [f'text: {k.get("corpus_doc")} score: {k.get("score"):.4f}' for k in res[0]])
     return res_show
 
 
@@ -32,12 +37,12 @@ if __name__ == '__main__':
         ['她在看书'],
         ['一个人弹琴'],
     ]
-
-    gr.Interface(ai_text,
-                 inputs=gr.Textbox(label="Enter Query"),
-                 outputs=gr.Textbox(label="Output Box"),
-                 title="Chinese Text Semantic Search Model",
-                 description="Copy or input Chinese text here. Submit and the machine will find the most similarity texts.",
-                 article="Link to <a href='https://github.com/shibing624/similarities' style='color:blue;' target='_blank\'>Github REPO: similarities</a>",
-                 examples=examples
-                 ).launch()
+    gr.Interface(
+        ai_text,
+        inputs=gr.Textbox(lines=2, label="Enter Query"),
+        outputs=gr.Textbox(label="Output Box"),
+        title="Chinese Text Semantic Search Model",
+        description="Copy or input Chinese text here. Submit and the machine will find the most similarity texts.",
+        article="Link to <a href='https://github.com/shibing624/similarities'  style='color:blue;' target='_blank'>Github REPO</a>",
+        examples=examples
+    ).launch()