inoki-giskard
commited on
Commit
•
2ecd2bd
1
Parent(s):
29d69db
Use popen to have a reference on ML worker process
Browse files
app.py
CHANGED
@@ -21,13 +21,15 @@ ml_worker = None
|
|
21 |
def run_ml_worker(url, api_key, hf_token):
|
22 |
global ml_worker, previous_url
|
23 |
previous_url = url
|
24 |
-
ml_worker = subprocess.
|
25 |
[
|
26 |
"giskard", "worker", "start",
|
27 |
"-u", f"{url}", "-k", f"{api_key}", "-t", f"{hf_token}"
|
28 |
],
|
29 |
stdout=open(LOG_FILE, "w"), stderr=subprocess.STDOUT
|
30 |
)
|
|
|
|
|
31 |
|
32 |
|
33 |
def stop_ml_worker():
|
|
|
21 |
def run_ml_worker(url, api_key, hf_token):
|
22 |
global ml_worker, previous_url
|
23 |
previous_url = url
|
24 |
+
ml_worker = subprocess.Popen(
|
25 |
[
|
26 |
"giskard", "worker", "start",
|
27 |
"-u", f"{url}", "-k", f"{api_key}", "-t", f"{hf_token}"
|
28 |
],
|
29 |
stdout=open(LOG_FILE, "w"), stderr=subprocess.STDOUT
|
30 |
)
|
31 |
+
args = ml_worker.args
|
32 |
+
print(f"Process {args} exited with {ml_worker.wait()}")
|
33 |
|
34 |
|
35 |
def stop_ml_worker():
|