Spaces:
Running
Running
Update service_config.py
Browse files- service_config.py +1 -1
service_config.py
CHANGED
@@ -23,7 +23,7 @@ class ServiceConfig:
|
|
23 |
self.LABEL2ID = {}
|
24 |
self.INTENT_DEFINITIONS = {}
|
25 |
|
26 |
-
def setup_environment():
|
27 |
os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "true"
|
28 |
os.environ["TORCH_HOME"] = "/app/.torch_cache"
|
29 |
os.makedirs("/app/.torch_cache", exist_ok=True)
|
|
|
23 |
self.LABEL2ID = {}
|
24 |
self.INTENT_DEFINITIONS = {}
|
25 |
|
26 |
+
def setup_environment(self):
|
27 |
os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "true"
|
28 |
os.environ["TORCH_HOME"] = "/app/.torch_cache"
|
29 |
os.makedirs("/app/.torch_cache", exist_ok=True)
|