Updated function name to huggingface_model_uri

status_check.py

@@ -8,7 +8,7 @@ from typing import Optional, Tuple
 
 from ui.statusui import StatusUI
 from checks.health_check import check_model_endpoint
-from agents.model import load_huggingface_model
+from agents.model import huggingface_model_uri
 
 
 def wake_up_endpoint(
@@ -96,7 +96,7 @@ def run_status_checks():
     ui.append("Starting prechecks...")
     ui.append("Checking endpoint..")
     
-    endpoint_uri = load_huggingface_model() # Get the URI for the endpoint 
+    endpoint_uri = huggingface_model_uri() # Get the URI for the endpoint 
     ui.append(endpoint_uri)   
 
     # Wake it up before health check