OK

src/draft_model.py

@@ -17,7 +17,6 @@ CARD_FILE = "cards_eoe.json"
 ENCODING_FILE = "card_encodings.pt"
 
 
-os.environ.setdefault("HF_HOME", os.path.expanduser("~/.cache/huggingface"))
 
 
 class DraftModel:

src/streamlit_app.py

@@ -14,14 +14,17 @@ for Hugging Face Spaces deployment.
 
 from __future__ import annotations
 
-import random
+import random, os
 from pathlib import Path
 from typing import Dict, List
 
 import requests
 import streamlit as st
 
-from draft_model import DraftModel
+from src.draft_model import DraftModel
+
+
+os.environ.setdefault("HF_HOME", os.path.expanduser("~/.cache/huggingface"))
 
 
 SUPPORTED_SETS_PATH = Path("supported_sets.txt")