Spaces:
Runtime error
Runtime error
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 |
|