RefalMachine commited on
Commit
b035ea0
·
verified ·
1 Parent(s): 49f1027

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -10,14 +10,21 @@ import requests
10
  from openai import OpenAI, AsyncOpenAI
11
 
12
  clients = {}
13
-
14
  try:
15
- clients['32B-Pro (beta)'] = [OpenAI(api_key==os.getenv('API_KEY'), base_url=os.getenv('RUADAPT_PRO_PATH')), requests.get(os.getenv('RUADAPT_PRO_PATH') + '/models').json()['data'][0]['id']]
 
 
 
 
16
  except:
17
  pass
18
 
19
  try:
20
- clients['32B QWQ (experimental, without any additional tuning after LEP!)'] = [OpenAI(api_key==os.getenv('API_KEY'), base_url=os.getenv('MODEL_NAME_OR_PATH_QWQ')), requests.get(os.getenv('MODEL_NAME_OR_PATH_QWQ') + '/models').json()['data'][0]['id']]
 
 
 
21
  except:
22
  pass
23
 
 
10
  from openai import OpenAI, AsyncOpenAI
11
 
12
  clients = {}
13
+ token = os.getenv('API_KEY')
14
  try:
15
+ clients['32B-Pro (beta)'] = [
16
+ OpenAI(
17
+ api_key==token,
18
+ base_url=os.getenv('RUADAPT_PRO_PATH')),
19
+ requests.get(os.getenv('RUADAPT_PRO_PATH') + '/models', headers={"Authorization": f"Bearer {token}"}).json()['data'][0]['id']]
20
  except:
21
  pass
22
 
23
  try:
24
+ clients['32B QWQ (experimental, without any additional tuning after LEP!)'] = [
25
+ OpenAI(api_key==token,
26
+ base_url=os.getenv('MODEL_NAME_OR_PATH_QWQ')),
27
+ requests.get(os.getenv('MODEL_NAME_OR_PATH_QWQ') + '/models', headers={"Authorization": f"Bearer {token}"}).json()['data'][0]['id']]
28
  except:
29
  pass
30