fixing tokenizer errors

app.py

@@ -2,11 +2,16 @@ from fastapi import FastAPI, HTTPException
 import chess
 from contextlib import asynccontextmanager
 import sys
-from src.model import ChessPolicyModel, PolicyModelInference 
-from src.tokenizer import tokenizer
 import torch
 from pydantic import BaseModel
 
+# Pickle stored the model/tokenizer classes under their original (top-level)
+# module paths. Alias src.* as top-level names so torch.load can resolve them.
+from src import tokenizer as _tokenizer_module
+sys.modules["tokenizer"] = _tokenizer_module
+
+from src.model import ChessPolicyModel, PolicyModelInference
+
 ml = {}
 @asynccontextmanager
 async def lifespan(app: FastAPI):