added middleware to note time

main.py

@@ -8,6 +8,7 @@ from typing import List, Union
 import numpy as np
 from arena import get_move_for_bot
 import torch
+import time
 
 class Request(BaseModel):
     board: List[List[int]]
@@ -38,6 +39,15 @@ model = Model(
 model.load_state_dict(torch.load(Config.checkpoint_path))
 model.eval()
 
+# Create a middleware to record time
+@app.middleware("http")
+async def add_process_time_header(req, call_next):
+    start_time = time.time()
+    response = await call_next(req)
+    process_time = time.time() - start_time
+    print(f'Time taken for response = {process_time:.2f} seconds')
+    return response
+
 @app.get("/")
 def root():
     return {"message": "This is a temporary response"}