Spaces:
Running
Running
File size: 3,281 Bytes
b4b0e11 |
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 102 103 104 105 106 107 108 109 110 111 |
#!/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()}")
# Verify torch version
if not torch.__version__.startswith('1.9'):
logger.warning(f"Expected PyTorch 1.9.x, got {torch.__version__}")
import detectron2
logger.info(f"Detectron2 version: {detectron2.__version__}")
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")
# Check numpy version
import numpy as np
logger.info(f"NumPy version: {np.__version__}")
return True
except ImportError as e:
logger.error(f"Missing dependency: {e}")
return False
def main():
"""Main application entry point"""
print("=" * 50)
print(f"NeuroNest Application Startup")
print(f"Time: {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=True,
debug = True
)
except Exception as e:
logger.error(f"Error launching app: {e}")
import traceback
traceback.print_exc()
sys.exit(1)
if __name__ == "__main__":
main()
|