Using spaces_info

main/main.py

@@ -7,7 +7,7 @@ from pathlib import Path
 from litgpt.api import LLM
 import os
 import uvicorn
-import whatismyip
+from huggingface_hub.hf_api import HfApi
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -170,7 +170,9 @@ def main():
     port = int(os.getenv("LLM_ENGINE_PORT", "8001"))
 
     # Start the server
-    logger.warning(f"Exposing IP {whatismyip.whatismyip()}")
+    hf_a = HfApi()
+    space_info = hf_a.space_info("TeamGenKI/LLM-Engine")
+    logger.warning(f"Exposing URL: {space_info}")
     uvicorn.run(
         app,
         host=host,