Update ruaccent/ruaccent.py

ruaccent/ruaccent.py

@@ -9,9 +9,9 @@ import re
 
 
 class RUAccent:
-    def __init__(self, workdir=None):
-        self.omograph_model = OmographModel()
-        self.accent_model = AccentModel()
+    def __init__(self, workdir=None, allow_cuda=True):
+        self.omograph_model = OmographModel(allow_cuda=allow_cuda)
+        self.accent_model = AccentModel(allow_cuda=allow_cuda)
         if not workdir:
             self.workdir = str(pathlib.Path(__file__).resolve().parent)
         else: