Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-#import logging
+import logging
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI, HTTPException
@@ -12,8 +12,8 @@ import torch.nn as nn
 from transformers import DistilBertModel, DistilBertTokenizer
 
 # ── Logging setup ─────────────────────────────────────────────────────────────
-#logging.basicConfig(level=logging.INFO)
-#logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # ── Label Maps ────────────────────────────────────────────────────────────────
 workout_label_map = {
@@ -72,9 +72,8 @@ state = AppState()
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # ── Startup ───────────────────────────────────────────────────────────────
-    print("Loading model, tokenizer and Supabase client...")
 
-    #logger.info("Loading model, tokenizer and Supabase client...")
+    logger.info("Loading model, tokenizer and Supabase client...")
 
     state.device = torch.device('cpu')
 
@@ -83,9 +82,8 @@ async def lifespan(app: FastAPI):
         'distilbert-base-uncased',
         token=os.getenv('HF_TOKEN')
     )
-    print("Tokenizer loaded")
 
-    #logger.info("Tokenizer loaded")
+    logger.info("Tokenizer loaded")
 
     # Load model once
     state.model = MultiHeadDistilBERT(