Upload 2 files

app.py

@@ -0,0 +1,56 @@
+from wiki_kb_qa_migrate import *
+
+'''
+zh_question = "哈利波特的血缘是什么？"
+#output = from_zh_question_to_consider_queries(zh_question)
+output = from_zh_question_to_consider_queries(zh_question,
+                                             top_k = 32, top_p_k = 30, top_e_k = 50
+                                             )
+if type(output) == type((1,)):
+    query_list, output = output
+    zh_output = trans_output(zh_question ,output)
+else:
+    zh_output = None
+zh_output
+ranking_output(zh_question, zh_output)
+'''
+
+import gradio as gr
+
+example_sample = [
+    "哈利波特的血缘是什么？",
+    "哈利波特的生日是什么？",
+    "占卜会议的领导者是谁？",
+    "纽约卫生局的创立时间是什么？",
+    "哥布林叛乱发生在什么日期？",
+    "赫敏的丈夫是谁？",
+]
+
+def demo_func(zh_question):
+    assert type(zh_question) == type("")
+    output = from_zh_question_to_consider_queries(zh_question,
+                                                 top_k = 32, top_p_k = 30, top_e_k = 50
+                                                 )
+    if type(output) == type((1,)):
+        query_list, output = output
+        zh_output = trans_output(zh_question ,output)
+    else:
+        zh_output = None
+    #zh_output
+    zh_output = ranking_output(zh_question, zh_output)
+    if hasattr(zh_output, "to_dict"):
+        zh_output = zh_output.head(5)
+        zh_output = zh_output[["answer", "sparql_query"]].values.tolist()
+    return {"output": zh_output}
+
+
+demo = gr.Interface(
+        fn=demo_func,
+        inputs="text",
+        outputs="json",
+        title=f"Harry Potter Knowledge Question Answer in Chinese 🧙 demonstration",
+        examples=example_sample if example_sample else None,
+        cache_examples = False,
+    )
+
+demo.launch(server_name=None, server_port=None)

requirements.txt

@@ -0,0 +1,11 @@
+editdistance==0.6.1
+fuzzywuzzy==0.18.0
+jionlp==1.4.7
+numpy==1.23.0
+pandas==1.4.3
+rapidfuzz==2.0.15
+synonyms==3.18.0
+torch
+tqdm==4.64.0
+"farm-haystack[inmemorygraph]"
+rdflib