inoki-giskard commited on
Commit
c0e96e4
1 Parent(s): 1bbdc1e

Avoid stop current running ML worker until exit

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -47,6 +47,7 @@ def read_status():
47
  def run_ml_worker(url, api_key, hf_token):
48
  global ml_worker, previous_url
49
  previous_url = url
 
50
  ml_worker = subprocess.Popen(
51
  [
52
  "giskard", "worker", "start",
@@ -74,6 +75,9 @@ def start_ml_worker(url, api_key, hf_token):
74
  if not url or len(url) < 1:
75
  return "Please provide URL of Giskard"
76
 
 
 
 
77
  # Always run an external ML worker
78
  stop_ml_worker()
79
 
 
47
  def run_ml_worker(url, api_key, hf_token):
48
  global ml_worker, previous_url
49
  previous_url = url
50
+ subprocess.run(["giskard", "worker", "stop"])
51
  ml_worker = subprocess.Popen(
52
  [
53
  "giskard", "worker", "start",
 
75
  if not url or len(url) < 1:
76
  return "Please provide URL of Giskard"
77
 
78
+ if ml_worker is not None:
79
+ return f"ML worker is still running for {previous_url}"
80
+
81
  # Always run an external ML worker
82
  stop_ml_worker()
83