Spaces:
Running
Running
File size: 2,268 Bytes
787c60a 937d315 787c60a 937d315 787c60a 937d315 787c60a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
"""
RoutePilot - GNN-LLM Intelligent Selection System
Hugging Face Spaces Entry Point
This is the main entry point for the Hugging Face Spaces deployment.
It imports the necessary components from the existing application and
sets up the Gradio interface for deployment.
"""
import os
import sys
import gradio as gr
# Add the current directory to Python path for imports
sys.path.append(os.path.dirname(__file__))
# Import the main application components
from demo import create_interface
# Set up environment variables for Hugging Face Spaces
def setup_environment():
"""Set up environment variables for Hugging Face Spaces deployment"""
# Check if we're running on Hugging Face Spaces
if os.getenv("SPACE_ID"):
print("🚀 Running on Hugging Face Spaces")
# Check for NVIDIA API key
if not os.getenv("NVIDIA_API_KEY"):
print("⚠️ NVIDIA_API_KEY not set in Space secrets.")
print(" Please set NVIDIA_API_KEY in the Space Repository Secrets.")
print(" Some features may be limited without API access.")
else:
print("✅ NVIDIA_API_KEY found in Space secrets")
# Set CUDA device for Spaces (usually limited resources)
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# Set memory optimization flags
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
else:
print("🏠 Running locally")
# Check for local .env file
if not os.getenv("NVIDIA_API_KEY"):
print("ℹ️ NVIDIA_API_KEY not found. For local development, create a .env file")
print(" or set the environment variable manually.")
def main():
"""Main function to launch the application"""
# Set up environment
setup_environment()
# Create the Gradio interface
print("🎯 Creating RoutePilot interface...")
demo = create_interface()
# Launch the application
# For Hugging Face Spaces, we don't need to specify server_name and port
# The platform handles this automatically
demo.launch(
show_error=True,
debug=False, # Set to False for production
show_api=False
)
if __name__ == "__main__":
main() |