cocktailpeanut commited on
Commit
ff70b56
·
1 Parent(s): dcc9adb
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -12,11 +12,11 @@ import tempfile
12
  import shlex
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
- MODEL = os.environ.get("MODEL", "qwen2.5-coder:32b-instruct-q4_0")
18
 
19
- client = OpenAI(base_url=BASE_URL, api_key=HF_API_KEY)
20
 
21
  allowed_medias = [
22
  ".png",
 
12
  import shlex
13
  import shutil
14
 
15
+ OPENAI_API_KEY = os.environ["OPENAI_API_KEY", "ollama"]
16
+ BASE_URL = os.environ.get("BASE_URL", "https://api.openai.com/v1/")
17
+ MODEL = os.environ.get("MODEL", "gpt-4o")
18
 
19
+ client = OpenAI(base_url=BASE_URL, api_key=OPENAI_API_KEY)
20
 
21
  allowed_medias = [
22
  ".png",