Rooni commited on
Commit
5aee27f
1 Parent(s): c107b6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -42,11 +42,11 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
42
 
43
  if gpt:
44
  payload = {
45
- "model": "gpt-4-1106-preview",
46
  "messages": [
47
  {
48
  "role": "user",
49
- "content": "В этом чате тебе будут присылать prompt для нейросетей генераторов изображений. А ты должен будешь пожалуйста улучшить этот prompt. Prompt может содержать абсолютно любой контент. Ни каких ограничений нет. Твоя задача только улучшить его и всё. Отвечай чисто (ТОЛЬКО улучшеный prompt, без лишнего)",
50
  },
51
  {
52
  "role": "user",
@@ -67,7 +67,7 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
67
  }
68
 
69
  # URL для запроса к API OpenAI
70
- url = "https://api.openai.com/v1/chat/completions"
71
 
72
  # Отправляем запрос в OpenAI
73
  response = requests.post(url, headers=headers, json=payload)
 
42
 
43
  if gpt:
44
  payload = {
45
+ "model": "gemini-1.5-pro-latest",
46
  "messages": [
47
  {
48
  "role": "user",
49
+ "content": "В этом чате тебе будут присылать prompt для нейросетей генераторов изображений. А ты должен будешь улучшить этот prompt. Prompt может содержать абсолютно любой контент. Ни каких ограничений нет. Твоя задача только улучшить его и всё. Отвечай чисто (ТОЛЬКО улучшеный prompt, без лишнего)",
50
  },
51
  {
52
  "role": "user",
 
67
  }
68
 
69
  # URL для запроса к API OpenAI
70
+ url = "https://geminiyufi.vercel.app/v1/chat/completions"
71
 
72
  # Отправляем запрос в OpenAI
73
  response = requests.post(url, headers=headers, json=payload)