Update app.py

app.py

@@ -21,13 +21,12 @@ sim_model.add_corpus(load_file('corpus.txt'))
 def ai_text(query):
     res = sim_model.most_similar(queries=query, topn=5)
     print(res)
-    for q_id, c in res.items():
-        print('query:', query)
-        print("search top 5:")
-        for corpus_id, s in c.items():
-            print(f'\t{sim_model.corpus[corpus_id]}: {s:.4f}')
-    res_show = '\n'.join(['search top5：'] + [f'text: {sim_model.corpus[corpus_id]} score: {s:.4f}' for corpus_id, s in
-                                             list(res.values())[0].items()])
+    for q_id, c in enumerate(res):
+        print('query:', sentences[q_id])
+        print("search top 3:")
+        print(f'\t{c}')
+
+    res_show = str(res)
     return res_show
 
 
@@ -44,6 +43,6 @@ if __name__ == '__main__':
                  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>",
+                 article="Link to <a href='https://github.com/shibing624/similarities' style='color:blue;' target='_blank\'>Github REPO: similarities</a>",
                  examples=examples
                  ).launch()