bgk commited on
Commit
5e7a6e2
·
1 Parent(s): 7355802

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ def ner(text):
5
+ model = "bgk/siparis"
6
+ trans_table = text.maketrans('', '', string.punctuation)
7
+ text = text.translate(trans_table)
8
+
9
+ text=text.lower()
10
+
11
+ prediction, model_output = model.predict([text])
12
+ filtered_output = [[{v: k} for d in sublist for k, v in d.items() if v.startswith("B-")] for sublist in prediction]
13
+ entities = []
14
+ for sublist in filtered_output:
15
+ for d in sublist:
16
+ for k, v in d.items():
17
+ label = k.split("-")[1]
18
+ entities.append((label, v))
19
+ return entities
20
+
21
+ demo = gr.Interface(fn=ner, inputs=gr.Textbox(lines=5, placeholder='Enter your input here'), outputs=gr.Textbox(lines=10, placeholder='Sipariş dökümü'),examples=['İki pide ve iki büyük ayran istiyorum, kapıda nakit öderim, evime yollayın...' , 'Bir mantı, iki döner, üç mevsim salata, bir kola ve iki sprite istiyorum, iş yerime gönderin' ],
22
+ title="Söyle Gelsin", description="Yemek siparişi uygulamamızı kullanabilirsiniz...")
23
+
24
+ demo.launch()