P-Karthik-Mohan commited on
Commit
7a189b8
·
1 Parent(s): fe5c216

Restored load_dotenv :)

Browse files
Files changed (1) hide show
  1. inference.py +3 -1
inference.py CHANGED
@@ -7,6 +7,8 @@ import json
7
  import time
8
  import requests
9
  from openai import OpenAI
 
 
10
 
11
  ENV_BASE_URL = "https://p-karthik-mohan-sql-analyst-env.hf.space"
12
  MAX_ATTEMPTS = 5
@@ -18,8 +20,8 @@ HF_TOKEN = os.environ.get("HF_TOKEN","")
18
 
19
  client = OpenAI(
20
  base_url=API_BASE_URL,
21
-
22
  api_key=HF_TOKEN if HF_TOKEN else "no-key-needed",
 
23
  )
24
 
25
  def env_reset(task_id: int) -> dict:
 
7
  import time
8
  import requests
9
  from openai import OpenAI
10
+ from dotenv import load_dotenv
11
+ load_dotenv()
12
 
13
  ENV_BASE_URL = "https://p-karthik-mohan-sql-analyst-env.hf.space"
14
  MAX_ATTEMPTS = 5
 
20
 
21
  client = OpenAI(
22
  base_url=API_BASE_URL,
 
23
  api_key=HF_TOKEN if HF_TOKEN else "no-key-needed",
24
+
25
  )
26
 
27
  def env_reset(task_id: int) -> dict: