File size: 2,942 Bytes
6524e7a 4e58082 6524e7a 8332d88 4e58082 6524e7a 4e58082 8332d88 6524e7a 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 8332d88 4e58082 6524e7a 8332d88 |
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 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
#!/usr/bin/env python3
"""
NeuroNest Application Entry Point
Handles initialization and graceful startup for Hugging Face Spaces
"""
import os
import sys
import logging
import time
from pathlib import Path
# Configure logging
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)
# Set environment variables
os.environ['CUDA_VISIBLE_DEVICES'] = ''
os.environ['FORCE_CUDA'] = '0'
def setup_oneformer_imports():
"""Add OneFormer to Python path if needed"""
oneformer_path = Path(__file__).parent / "oneformer"
if oneformer_path.exists() and str(oneformer_path) not in sys.path:
sys.path.insert(0, str(oneformer_path))
logger.info(f"Added OneFormer to path: {oneformer_path}")
def check_dependencies():
"""Check if all required dependencies are available"""
try:
import torch
logger.info(f"PyTorch version: {torch.__version__}")
logger.info(f"CUDA available: {torch.cuda.is_available()}")
import detectron2
logger.info(f"Detectron2 imported successfully")
import gradio as gr
logger.info(f"Gradio version: {gr.__version__}")
import cv2
logger.info(f"OpenCV version: {cv2.__version__}")
import PIL
logger.info(f"Pillow version: {PIL.__version__}")
# Check PIL compatibility
if hasattr(PIL.Image, 'LINEAR'):
logger.info("PIL has LINEAR attribute")
elif hasattr(PIL.Image, 'BILINEAR'):
logger.info("PIL has BILINEAR attribute (newer version)")
# Monkey patch for compatibility
PIL.Image.LINEAR = PIL.Image.BILINEAR
logger.info("Applied PIL compatibility patch")
return True
except ImportError as e:
logger.error(f"Missing dependency: {e}")
return False
def main():
"""Main application entry point"""
print("=" * 50)
print(f"Application Startup at {time.strftime('%Y-%m-%d %H:%M:%S')}")
print("=" * 50)
# Setup paths
setup_oneformer_imports()
# Check dependencies
if not check_dependencies():
logger.error("Dependency check failed")
sys.exit(1)
try:
# Import and launch the Gradio interface
from gradio_test import create_gradio_interface
logger.info("Creating Gradio interface...")
interface = create_gradio_interface()
logger.info("Launching application...")
interface.queue(max_size=10).launch(
server_name="0.0.0.0",
server_port=7860,
share=False # Disable share for production
)
except Exception as e:
logger.error(f"Error: {e}")
import traceback
traceback.print_exc()
sys.exit(1)
if __name__ == "__main__":
main()
|