Spaces:
Build error
Build error
Update src/Surveyor.py
Browse files- src/Surveyor.py +3 -3
src/Surveyor.py
CHANGED
@@ -87,9 +87,9 @@ class Surveyor:
|
|
87 |
spacy.require_gpu()
|
88 |
|
89 |
self.high_gpu = high_gpu
|
90 |
-
self.DEFAULTS = DEFAULTS_CPU_COMPAT
|
91 |
-
if self.high_gpu:
|
92 |
-
|
93 |
|
94 |
if not kw_model_name:
|
95 |
kw_model_name = self.DEFAULTS["kw_model_name"]
|
|
|
87 |
spacy.require_gpu()
|
88 |
|
89 |
self.high_gpu = high_gpu
|
90 |
+
#self.DEFAULTS = DEFAULTS_CPU_COMPAT
|
91 |
+
#if self.high_gpu:
|
92 |
+
self.DEFAULTS = DEFAULTS_HIGH_GPU
|
93 |
|
94 |
if not kw_model_name:
|
95 |
kw_model_name = self.DEFAULTS["kw_model_name"]
|