cocktailpeanut commited on
Commit
80d34f7
1 Parent(s): 9d2de5c
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -13,8 +13,10 @@ import shlex
13
  import shutil
14
 
15
  HF_API_KEY = os.environ["HF_TOKEN"]
 
16
 
17
- client = OpenAI(base_url="https://api-inference.huggingface.co/v1/", api_key=HF_API_KEY)
 
18
 
19
  allowed_medias = [
20
  ".png",
 
13
  import shutil
14
 
15
  HF_API_KEY = os.environ["HF_TOKEN"]
16
+ BASE_URL = os.environ.get("BASE_URL", "https://api-inference.huggingface.co/v1/")
17
 
18
+
19
+ client = OpenAI(base_url=BASE_URL, api_key=HF_API_KEY)
20
 
21
  allowed_medias = [
22
  ".png",