weird fixes

src/draft_model.py

@@ -4,9 +4,7 @@ import torch
 from typing import List, Dict
 
 from huggingface_hub import hf_hub_download
-
-from src.training import train_mlp
-from src.utils import utils
+from helpers import get_embedding_dict, get_card_embeddings, MLP_CrossAttention
 
 
 MODEL_REPO = "TimoBertram/MTG_Model_FIN/"
@@ -18,7 +16,6 @@ DATA_REPO = "TimoBertram/MTG_Drafting_Dataset/"
 CARD_FILE = "cards_eoe.json"
 ENCODING_FILE = "card_encodings.pt"
 
-from helpers import get_embedding_dict, get_card_embeddings, MLP_CrossAttention