fix bugs

app.py

@@ -10,17 +10,17 @@ from transformers import pipeline
 import spacy
 from spacy import displacy
 
-POS_TAG_MAP = {0: "PODMET", 16: "PRÍSUDOK", 6: "PRÍDAVNÉ MENO", 2: "ADPOZÍCIA",
-               14: "PRÍSLOVKA", 17: "POMOCNÉ", 9: "KORDINAČNÁ SPOJKA", 8: "DETERMINANT",
-               15: "CYTOSLOVCIA", 3: "NUMERICKÉ", 7: "ČASTICA", 11: "ZÁMMENO",
-               10: "VLASTNÉ MENO", 1: "INTERPUNKCIA", 5: "PODRAĎOVACIA SPOJKA", 4: "SYMBÓL",
-               12: "INÉ", 13: "_"}
+
+POS_TAG_MAP = {"NOUN": "PODMET", "VERB": "PRÍSUDOK", "ADJ": "PRÍDAVNÉ MENO", "ADP": "ADPOZÍCIA",
+               "ADV": "PRÍSLOVKA", "AUX": "POMOCNÉ", "CCONJ": "KORDINAČNÁ KONJUKCIA", "DET": "DETERMINANT",
+               "INTJ": "CYTOSLOVCIA", "NUM": "NUMERICKÉ", "PART": "ČASTICA", "PRON": "ZÁMMENO",
+               "PROPN": "VLASTNÉ MENO", "PUNCT": "INTERPUNKCIA", "SCONJ": "", "SYM": "SYMBÓL", "X": "INÉ"}
 
 options = {"ents": list(POS_TAG_MAP.values()),
            "colors": {"PODMET": "lightblue", "PRÍSUDOK": "lightcoral",
                       "PRÍDAVNÉ MENO": "lightgreen", "VLASTNÉ MENO": "papayawhip"}}
 
-pipe = pipeline(task='token-classification', model="crabz/slovakbert-ner")
+pipe = pipeline(task='token-classification', model="crabz/slovakbert-upos")
 nlp = spacy.blank("sk")
 
 
@@ -55,19 +55,14 @@ def set_entities(sentence, entities):
     for label, start, end in entities:
         ents.append(doc.char_span(start, end, label))
     try:
-        print(ents)
         doc.ents = ents
-    except TypeError as e:
-        print(e)
+    except TypeError:
+        pass
     return doc
 
 
 def apply_pos(Sentence: str):
-    classifications = pipe(Sentence.replace(",", " ,")
-                           .replace(".", " .")
-                           .replace("!", " !")
-                           .replace("?", " ?")
-                           .replace(":", " :"))
+    classifications = pipe(Sentence)
     entities = postprocess(classifications)
     print(entities)
     doc = set_entities(Sentence, entities)
@@ -85,4 +80,4 @@ intf = gr.Interface(fn=apply_pos, inputs="text", outputs="html", title='Slovak P
                     description="",
                     article="",
                     analytics_enabled=False)
-intf.launch()
+intf.launch()