23 lines
495 B
Python
23 lines
495 B
Python
|
import os
|
||
|
|
import requests
|
||
|
|
|
||
|
|
inference_address = os.environ["INFERENCE_API_ADDRESS"]
|
||
|
|
url = f"{inference_address}/update"
|
||
|
|
|
||
|
|
print("UPDATING INFERENCE WORKER DATA")
|
||
|
|
|
||
|
|
response = requests.get(url)
|
||
|
|
if response.status_code == 200:
|
||
|
|
# Request was successful
|
||
|
|
content = response.text
|
||
|
|
|
||
|
|
if content == "0":
|
||
|
|
print("Response content is '0'")
|
||
|
|
exit(0)
|
||
|
|
else:
|
||
|
|
exit(1)
|
||
|
|
else:
|
||
|
|
# Request failed
|
||
|
|
print(f"Request failed with status code: {response.status_code}")
|
||
|
|
exit(1)
|