Update WorkerNode.py

uplion · Jul 19, 2024 · 1821781 · 1821781
1 parent 430d3a9
commit 1821781
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/WorkerNodePython/WorkerNode/WorkerNode.py b/WorkerNodePython/WorkerNode/WorkerNode.py
@@ -16,6 +16,8 @@
 
 load_dotenv()
 
+AIModelNamespace = os.getenv('AIMODEL_NAMESPACE')
+AIModelName = os.getenv('AIMODEL_NAME')
 nodeType : str = '';
 pulsarURL : str = '';
 topicName : str = '';
@@ -269,4 +271,4 @@ def sendResult(self,result):
 
 if __name__ == '__main__':
     init()
-    run()
+    run()