new version

src/category.py

@@ -43,7 +43,8 @@ class Model:
                 self.synsets[str(synset)] = (tag, lemma)
             else:
                 if var is not None:
-                    self.variants[lemma.replace(" ","_") + "." + str(var)] = (tag, lemma)
+                    t = lemma.replace(" ", "_") + "." + str(var)
+                    self.variants[t] = (tag, lemma)
                 else:
                     self.lemmas[lemma] = tag
         wb.close()