Spaces:
Sleeping
Sleeping
Update status_check.py
Browse files- status_check.py +0 -6
status_check.py
CHANGED
|
@@ -3,9 +3,6 @@ import time
|
|
| 3 |
import requests
|
| 4 |
|
| 5 |
from typing import Optional, Tuple
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
from ui.statusui import StatusUI
|
| 10 |
from checks.health_check import check_model_endpoint
|
| 11 |
from agents.model import huggingface_model_uri
|
|
@@ -92,17 +89,14 @@ def run_status_checks():
|
|
| 92 |
"""Run all status checks and return endpoint URI if successful"""
|
| 93 |
ui = StatusUI("Content Agent Status Checks")
|
| 94 |
ui.launch()
|
| 95 |
-
|
| 96 |
ui.append("Starting prechecks...")
|
| 97 |
ui.append("Checking endpoint..")
|
| 98 |
-
|
| 99 |
endpoint_uri = huggingface_model_uri() # Get the URI for the endpoint
|
| 100 |
ui.append(endpoint_uri)
|
| 101 |
|
| 102 |
# Wake it up before health check
|
| 103 |
wake_up_successful = wake_up_endpoint(endpoint_uri, ui)
|
| 104 |
success, error_msg = wake_up_endpoint(endpoint_uri, ui)
|
| 105 |
-
i
|
| 106 |
if not success:
|
| 107 |
ui.append("Warning: Could not wake up the endpoint. Exiting.")
|
| 108 |
|
|
|
|
| 3 |
import requests
|
| 4 |
|
| 5 |
from typing import Optional, Tuple
|
|
|
|
|
|
|
|
|
|
| 6 |
from ui.statusui import StatusUI
|
| 7 |
from checks.health_check import check_model_endpoint
|
| 8 |
from agents.model import huggingface_model_uri
|
|
|
|
| 89 |
"""Run all status checks and return endpoint URI if successful"""
|
| 90 |
ui = StatusUI("Content Agent Status Checks")
|
| 91 |
ui.launch()
|
|
|
|
| 92 |
ui.append("Starting prechecks...")
|
| 93 |
ui.append("Checking endpoint..")
|
|
|
|
| 94 |
endpoint_uri = huggingface_model_uri() # Get the URI for the endpoint
|
| 95 |
ui.append(endpoint_uri)
|
| 96 |
|
| 97 |
# Wake it up before health check
|
| 98 |
wake_up_successful = wake_up_endpoint(endpoint_uri, ui)
|
| 99 |
success, error_msg = wake_up_endpoint(endpoint_uri, ui)
|
|
|
|
| 100 |
if not success:
|
| 101 |
ui.append("Warning: Could not wake up the endpoint. Exiting.")
|
| 102 |
|