Rooni commited on
Commit
cb15c7e
·
verified ·
1 Parent(s): daec028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -23,13 +23,12 @@ def generate(description, system_role_name, max_tokens):
23
 
24
  headers = {
25
  'Content-Type': 'application/json',
26
- 'Authorization': f'Bearer {os.getenv("API_KEY")}'
27
  }
28
 
29
  payload = {
30
  'messages': [{'role': 'system', 'content': system_role}, {'role': 'user', 'content': description}],
31
  'max_tokens': max_tokens,
32
- 'model': "gemini-2.0-flash",
33
  'stream': True
34
  }
35
 
@@ -65,7 +64,7 @@ css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
65
 
66
  # Получение CSS по ссылке
67
  response = requests.get(css_url)
68
- css = response.text + ".gradio-container{max-width: 700px !important} h1{text-align:center}"
69
 
70
  # Загрузка названий ролей из JSON файла
71
  role_names = load_role_names()
 
23
 
24
  headers = {
25
  'Content-Type': 'application/json',
 
26
  }
27
 
28
  payload = {
29
  'messages': [{'role': 'system', 'content': system_role}, {'role': 'user', 'content': description}],
30
  'max_tokens': max_tokens,
31
+ 'model': "openai",
32
  'stream': True
33
  }
34
 
 
64
 
65
  # Получение CSS по ссылке
66
  response = requests.get(css_url)
67
+ css = response.text + ".gradio-container{max-width: 700px !important; margin: 0 auto;} h1{text-align: center;}"
68
 
69
  # Загрузка названий ролей из JSON файла
70
  role_names = load_role_names()