AK1239 commited on
Commit
8d0cfc1
·
1 Parent(s): 725e4de

fixed logger issue

Browse files
Files changed (1) hide show
  1. app/main.py +6 -4
app/main.py CHANGED
@@ -414,16 +414,18 @@ logger = logging.getLogger(__name__)
414
  # Initialize global variables in app state
415
  @app.on_event("startup")
416
  async def startup_event():
417
- # Check if CUDA is available
 
 
 
 
 
418
  device = "cuda" if torch.cuda.is_available() else "cpu"
419
  logger.info(f"Using device: {device}")
420
 
421
  if device == "cpu":
422
  logger.warning("GPU not detected. Model will run slower on CPU.")
423
 
424
- """Initialize the application on startup."""
425
- logger = logging.getLogger(__name__)
426
- logger.info("Starting application initialization...")
427
 
428
  # Set NLTK data path
429
  nltk_data_dir = os.environ.get('NLTK_DATA', os.path.join(os.path.expanduser('~'), 'nltk_data'))
 
414
  # Initialize global variables in app state
415
  @app.on_event("startup")
416
  async def startup_event():
417
+
418
+ """Initialize the application on startup."""
419
+ logger = logging.getLogger(__name__)
420
+ logger.info("Starting application initialization...")
421
+
422
+ # Check if CUDA is available
423
  device = "cuda" if torch.cuda.is_available() else "cpu"
424
  logger.info(f"Using device: {device}")
425
 
426
  if device == "cpu":
427
  logger.warning("GPU not detected. Model will run slower on CPU.")
428
 
 
 
 
429
 
430
  # Set NLTK data path
431
  nltk_data_dir = os.environ.get('NLTK_DATA', os.path.join(os.path.expanduser('~'), 'nltk_data'))