Update sft_fastapi.py

sft_fastapi.py

@@ -7,7 +7,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import random
 import os
 from typing import List, Optional
-from sft_dataset import load_config
 
 app = FastAPI(title="PVS Step Recommender API", version="1.0.0")
 
@@ -19,6 +18,12 @@ MODEL = None
 TEST_DATASET = None
 DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
 
+def load_config(path="config.yaml"):
+    with open(path, "r") as f:
+        cfg = yaml.safe_load(f)
+    return SimpleNamespace(**cfg)
+
+
 def load_model_and_tokenizer(path: str):
     global TOKENIZER, MODEL
     if TOKENIZER is None or MODEL is None: