Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+"""import gradio as gr
 from simpletransformers.ner import NERModel 
 import string
 
@@ -15,8 +15,61 @@ def ner(text, model):
     prediction, model_output = model.predict([text])
     output = prediction
     return output
-text = inputs
+
 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' ],
     title="Söyle Gelsin", description="Yemek siparişi uygulamamızı kullanabilirsiniz...")
 
-demo.launch()
+demo.launch()
+"""
+
+from simpletransformers.ner import NERModel 
+import string
+import gradio as gr
+
+def ner(text):
+    model = NERModel("albert", "bgk/lodosalberttr",use_cuda=False)
+    trans_table = text.maketrans('', '', string.punctuation)
+    text = text.translate(trans_table)
+
+    text=text.lower()
+    
+    prediction, model_output = model.predict([text])
+    filtered_output = [[{v: k} for d in sublist for k, v in d.items() if v.startswith("B-")] for sublist in prediction]
+    entities = []
+    for sublist in filtered_output:
+        for d in sublist:
+            for k, v in d.items():
+                label = k.split("-")[1]
+                entities.append((label, v))
+    return entities
+
+demo = gr.Interface(fn=ner, inputs=gr.Textbox(lines=5, placeholder='Enter your input here'), outputs=gr.Textbox(lines=5, placeholder='Sipariş dökümü'),examples=['Bize iki buçuk pide ve iki büyük ayran istiyorum, kapıda nakit öderim, evime yollayın...' , 'bir mantı, bir döner, bir mevsim salata, bir kola ve 1 sprite istiyorum' ],
+    title="Söyle Gelsin", description="Yemek siparişi uygulamamızı kullanabilirsiniz...")
+
+demo.launch()
+""" 
+model = NERModel(
+        "albert", 
+        "bgk/lodosalberttr", ignore_mismatched_sizes=True,
+        use_cuda=False
+    )
+
+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"],
+    ["bana adana kebap iki adet, yanına bir kola ve ayran ver bir tane"]
+            
+]
+
+    
+def ner(text):
+    output = ner_pipeline(text)
+    return {"text": text, "entities": output}    
+
+demo = gr.Interface(ner,
+             gr.Textbox(placeholder="Inserisci una frase qui..."), 
+             gr.HighlightedText(),
+             examples=examples)
+
+demo.launch()
+"""