vilarin commited on
Commit
d0a1d12
·
verified ·
1 Parent(s): 1ac84a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,9 +5,9 @@ import os
5
  import base64
6
 
7
  # Set API key and organization ID from environment variables
8
- openai.api_key = os.getenv('OPENAI_API_KEY')
9
- openai.organization = os.getenv('OPENAI_ORG_ID')
10
- client = OpenAI(api_key= os.getenv('OPENAI_API_KEY'), organization=os.getenv('OPENAI_ORG_ID'))
11
 
12
  # Define the model to be used
13
  MODEL = "gpt-4o"
 
5
  import base64
6
 
7
  # Set API key and organization ID from environment variables
8
+ api_key = os.environ.get("OPENAI_API_KEY")
9
+ base_url = os.environ.get("OPENAI_API_BASE")
10
+ client = OpenAI(api_key=api_key, base_url=base_url)
11
 
12
  # Define the model to be used
13
  MODEL = "gpt-4o"