yetessam commited on
Commit
907067c
·
verified ·
1 Parent(s): ae5b26b

Update status_check.py

Browse files
Files changed (1) hide show
  1. 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