Update inference.py

inference.py

@@ -28,7 +28,7 @@ load_dotenv()
 # ---------------------------------------------------------------------------
 # Configuration
 # ---------------------------------------------------------------------------
-
+"""
 API_KEY      = os.getenv("API_KEY", "")
 API_BASE_URL = os.getenv("API_BASE_URL")
 MODEL_NAME   = os.getenv("MODEL_NAME")
@@ -36,6 +36,16 @@ ENV_URL      = os.getenv("ENV_URL", "http://localhost:7860")
 BENCHMARK    = "xsecure"
 MAX_STEPS    = 20
 
+"""
+
+API_BASE_URL = os.environ.get("API_BASE_URL",  "https://api.openai.com/v1")
+MODEL_NAME   = os.environ.get("MODEL_NAME",  "gpt-4o-mini")
+API_KEY      = os.environ.get("API_KEY", "")
+ENV_URL      = os.environ.get("ENV_URL", "http://localhost:7860")
+
+BENCHMARK    = "xsecure"
+MAX_STEPS    = 20
+
 if not API_KEY:
     print("ERROR: HF_TOKEN is not set.", file=sys.stderr)
     sys.exit(1)