fixed logger issue
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
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'))
|