fix: relative file import paths

prediction.py

@@ -1,7 +1,7 @@
 import torch
 import tiktoken
-from constants.tokens import special_tokens
-from services.model import load_model, get_device
+from src.constants.tokens import special_tokens
+from src.services.model import load_model, get_device
 
 # Initialize tokenizer
 _tokenizer = tiktoken.get_encoding("cl100k_base")

src/services/model.py

@@ -2,7 +2,7 @@ import os
 import json
 import torch
 # from pathlib import Path
-from services.transformer import TinyTransformer
+from src.services.transformer import TinyTransformer
 
 # Internal constants for file paths
 # _MODEL_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)), "model")

src/services/transformer.py

@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-from constants.tokens import PAD_ID
+from src.constants.tokens import PAD_ID
 
 class TinyTransformer(nn.Module):
   def __init__(self, vocab_size, d_model=256, nhead=4, num_layers=2, dim_feedforward=512, dropout=0.1):