fix

app.py

@@ -100,7 +100,7 @@ lex = LexRank()
 def randeng_extract(content):
     summary_length = math.ceil(len(content) / 10)
     sentences = lex.find_central(content, num=summary_length)
-    output = "原文: \n"
+    output = ""
     for index, sentence in enumerate(sentences):
         output += f"{index}: {sentence}\n"
     # output += "摘要:\n"
@@ -135,9 +135,9 @@ def similarity_search(queries, doc):
         top_results = torch.topk(cos_scores, k=top_k)
         output += "\n\n======================\n\n"
         output += f"Query: {query}"
-        output += "\nTop 5 most similar sentences in corpus:"
+        output += "\nTop 5 most similar sentences in corpus:\n"
         for score, idx in zip(top_results[0], top_results[1]):
-            output += f"{doc_list[idx]}(Score: {score})"
+            output += f"{doc_list[idx]}(Score: {score})\n"
     return output
 
 
@@ -154,8 +154,8 @@ with gr.Blocks() as app:
     #     text_button = gr.Button("生成摘要")
     with gr.Tab("相似度检测"):
         with gr.Row():
-            text_input_query = gr.Textbox(label="查询文本")
-            text_input_doc = gr.Textbox(lines=10, label="逐行输入待比较的文本列表")
+            text_input_query = gr.Textbox(lines=10, label="查询文本")
+            text_input_doc = gr.Textbox(lines=20, label="逐行输入待比较的文本列表")
         text_button_similarity = gr.Button("对比相似度")
         text_output_similarity = gr.Textbox()