Spaces:
dxdcx
/
Running on CPU Upgrade

dxdcx commited on
Commit
d8fa89d
·
verified ·
1 Parent(s): b1beb36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -16,13 +16,14 @@ import spaces
16
  # Friendli AI Endpoints parameter
17
  friendli_token = os.getenv("FRIENDLI_TOKEN", "your_friendli_token")
18
  gemini_token = os.getenv("GEMINI_TOKEN", "your_gemini_token")
 
19
 
20
  # OpenAI client for Friendli
21
  client = OpenAI(
22
  # base_url="https://api.friendli.ai/dedicated/v1",
23
- # api_key=friendli_token
24
- base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
25
- api_key=gemini_token
26
  )
27
 
28
  MAX_NUM_IMAGES = int(os.getenv("MAX_NUM_IMAGES", "5"))
 
16
  # Friendli AI Endpoints parameter
17
  friendli_token = os.getenv("FRIENDLI_TOKEN", "your_friendli_token")
18
  gemini_token = os.getenv("GEMINI_TOKEN", "your_gemini_token")
19
+ openai_token = os.getenv("OPENAI_TOKEN", "your_openai_token")
20
 
21
  # OpenAI client for Friendli
22
  client = OpenAI(
23
  # base_url="https://api.friendli.ai/dedicated/v1",
24
+ # base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
25
+ base_url = "",
26
+ api_key=openai_token,
27
  )
28
 
29
  MAX_NUM_IMAGES = int(os.getenv("MAX_NUM_IMAGES", "5"))