Update app.py

app.py

@@ -25,9 +25,15 @@ import gradio as gr
 from simpletransformers.ner import NERModel 
 import string
 
+labels = ["B-YEMEK","O","B-ICECEK_BUYUKLUK","B-ICECEK","B-ADRES","B-ODEME","B-YEMEK_CESIT","B-YEMEK_ADET","B-ICECEK_ICERIK","I-ADRES","I-YEMEK","B-YEMEK_BUYUKLUK","B-YEMEK_ICERIK","I-YEMEK_ICERIK","I-ICECEK","I-YEMEK_BUYUKLUK","I-YEMEK_CESIT","B-ICECEK_ADET","I-ODEME","I-YEMEK_ADET"]
+model = NERModel(
+        "albert", 
+        "bgk/lodosalberttr", labels=labels,
+        use_cuda=False,
+        ignore_mismatched_sizes=True
+    )
 
 def ner(text):
-    model = NERModel("albert", "bgk/lodosalberttr",ignore_mismatched_sizes=True,use_cuda=False)
     trans_table = text.maketrans('', '', string.punctuation)
     text = text.translate(trans_table)