fix

app.py

@@ -3,10 +3,20 @@ from pydantic import BaseModel
 from typing import Any, Dict, List, Optional
 
 import json
+import sys
+from pathlib import Path
+
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel
 
+ROOT_DIR = Path(__file__).resolve().parent
+RL_DIR = ROOT_DIR / "RL"
+for path in (ROOT_DIR, RL_DIR):
+    path_str = str(path)
+    if path_str not in sys.path:
+        sys.path.append(path_str)
+
 from RL.battleground_nl_utils import game_state_to_natural_language