derek-thomas HF staff commited on
Commit
53724ab
1 Parent(s): e093898

Adding strong styling for non-running status

Browse files
Files changed (1) hide show
  1. backend/query_llm.py +2 -2
backend/query_llm.py CHANGED
@@ -63,9 +63,9 @@ def check_endpoint_status():
63
 
64
  # Extracting the status information
65
  status = data.get('status', {}).get('state', 'No status found')
66
- message = data.get('status', {}).get('message', 'No message found')
67
 
68
- if status == "scaledToZero":
69
  return f"<div style='color: red; font-size: 20px; font-weight: bold;'>Status: {status}<br>Message: {message}</div>"
70
  else:
71
  return f"<div>Status: {status}<br>Message: {message}</div>"
 
63
 
64
  # Extracting the status information
65
  status = data.get('status', {}).get('state', 'No status found')
66
+ message = data.get('status', {}).get('message', 'No message found').replace('Sending a request will restart the Endpoint', 'Click Wake Up Endpoint')
67
 
68
+ if status.lower() != "running":
69
  return f"<div style='color: red; font-size: 20px; font-weight: bold;'>Status: {status}<br>Message: {message}</div>"
70
  else:
71
  return f"<div>Status: {status}<br>Message: {message}</div>"