update tb and logger

requirements.txt

@@ -5,6 +5,7 @@ pydantic==1.10.7
 rich==13.7.0
 scikit_learn==1.1.2
 torch==2.0.1
+tensorboard==2.13.0
 transformers==4.28.0.dev0
 uvicorn==0.24.0.post1
 gpu-watchmen==0.3.8

src/app/api_backend.py

@@ -7,6 +7,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import FileResponse
 from pydantic import BaseModel
 from rex.utils.initialization import set_seed_and_log_path
+from rex.utils.logging import logger
 
 from src.task import SchemaGuidedInstructBertTask
 
@@ -54,6 +55,7 @@ def process_data(data: RequestData):
         msg = traceback.format_exc()
 
     # Return the processed data
+    logger.info(f"Data: {input_data}, Prediction: {results}")
     return {"ok": ok, "msg": msg, "results": results}