wjmcat commited on
Commit
8ed590c
·
1 Parent(s): 9279ebe

Create new file

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import paddlenlp
2
+ import gradio as gr
3
+
4
+ from paddlenlp import Taskflow
5
+ from paddlenlp.taskflow.text_similarity import TextSimilarityTask
6
+
7
+ def get_response():
8
+ similarity = Taskflow("text_similarity")
9
+ # query = "有哪些功能"
10
+ # support_set = ["你有哪些功能呢?", "你能做什么?", "你能帮我们做什么?"]
11
+
12
+ query = "春天适合种什么花?"
13
+ support_set = ["春天适合种什么菜?"]
14
+
15
+ inputs = []
16
+ for support_item in support_set:
17
+ inputs.append(
18
+ [query, support_item]
19
+ )
20
+
21
+ result = similarity(inputs)
22
+
23
+ def greet(name):
24
+ result = get_response()
25
+ return str(result)
26
+
27
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
28
+ iface.launch()