Update app.py

app.py

@@ -8,11 +8,7 @@ api_key = os.getenv("token")
 login(token = api_key)
 
 
-get_completion = pipeline("ner", model="elnasharomar2/PUNCERT_single_stages_70_epochs")
-label_names = ["O","QE","EX","QM","DOT","COM","SEMICOL","COL"]
-label_symbol = ["O",'؟!','!','؟','.','،','؛',':']
-
-id2label = {i: label for i, label in zip(label_names,label_symbol)}
+get_completion = pipeline("ner", model="elnasharomar2/PUNCERT_multi_stages_50_epochs")
 
 def merge_tokens(tokens):
     merged_tokens = []
@@ -32,7 +28,6 @@ def merge_tokens(tokens):
 def ner(input):
     output = get_completion(input)
     merged_tokens = merge_tokens(output)
-
     return {"text": input, "entities": merged_tokens}
 
 gr.close_all()