Spaces:
Sleeping
Sleeping
Commit ·
7a189b8
1
Parent(s): fe5c216
Restored load_dotenv :)
Browse files- 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:
|