update logging

serve.py

@@ -7,6 +7,11 @@ import numpy as np
 import io
 from pydub import AudioSegment
 import time
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model = load_model(device)
@@ -55,5 +60,5 @@ async def transcribe_endpoint(file: UploadFile = File(...)):
     
 
 if __name__ == "__main__":
-    print(f"Model loaded on {device}")
+    logger.info(f"Model loaded on {device}")
     uvicorn.run(app, host="0.0.0.0", port=7860)