Spaces:
Runtime error
Runtime error
Commit
·
629a811
1
Parent(s):
e05b343
Set ML worker to None after exiting
Browse files
app.py
CHANGED
@@ -44,6 +44,7 @@ def run_ml_worker(url, api_key, hf_token):
|
|
44 |
)
|
45 |
args = ml_worker.args
|
46 |
print(f"Process {args} exited with {ml_worker.wait()}")
|
|
|
47 |
|
48 |
|
49 |
def stop_ml_worker():
|
@@ -51,6 +52,7 @@ def stop_ml_worker():
|
|
51 |
if ml_worker is not None:
|
52 |
print(f"Stopping ML worker for {previous_url}")
|
53 |
ml_worker.terminate()
|
|
|
54 |
print("ML worker stopped")
|
55 |
return "ML worker stopped"
|
56 |
return "ML worker not started"
|
|
|
44 |
)
|
45 |
args = ml_worker.args
|
46 |
print(f"Process {args} exited with {ml_worker.wait()}")
|
47 |
+
ml_worker = None
|
48 |
|
49 |
|
50 |
def stop_ml_worker():
|
|
|
52 |
if ml_worker is not None:
|
53 |
print(f"Stopping ML worker for {previous_url}")
|
54 |
ml_worker.terminate()
|
55 |
+
ml_worker = None
|
56 |
print("ML worker stopped")
|
57 |
return "ML worker stopped"
|
58 |
return "ML worker not started"
|