Rooni commited on
Commit
21608b4
·
1 Parent(s): f0c75f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,14 +3,14 @@ import requests
3
  import json
4
  import os
5
 
6
- def generate_minecraft_command(description):
7
  headers = {
8
  'Content-Type': 'application/json',
9
  'Authorization': f'Bearer {os.getenv("API_KEY")}'
10
  }
11
 
12
  payload = {
13
- 'messages': [{'role': 'system', 'content': f'{description}'}],
14
  'max_tokens': 10000,
15
  'model': os.getenv("MODEL")
16
  }
@@ -27,7 +27,7 @@ def generate_minecraft_command(description):
27
  else:
28
  return f'Не удалось сгенерировать promt. {data}'
29
 
30
- iface = gr.Interface(fn=generate_minecraft_command, inputs=[
31
  gr.Textbox(label="Запрос")
32
  ], outputs=gr.Textbox(label="Ответ"), title="Генератор prompt для нейросетей")
33
  iface.launch()
 
3
  import json
4
  import os
5
 
6
+ def generate_promt(description):
7
  headers = {
8
  'Content-Type': 'application/json',
9
  'Authorization': f'Bearer {os.getenv("API_KEY")}'
10
  }
11
 
12
  payload = {
13
+ 'messages': [{'role': 'system', 'content': f'Сгенерируй пожалуйста promt для нейросети генерирующая изображения, соенерируй по описанию "{description}". Должно быть описание изображения (prompt), а не просьба/приказ. Нейросеть не знает не каких персонажей/построек/городов, по этому если там будет что нибудь из этого, то опиши это отлично и понятно. Напиши ТОЛЬКО promt, без дополнительного текста. Кавычки в начале и в конце не нужны.'}],
14
  'max_tokens': 10000,
15
  'model': os.getenv("MODEL")
16
  }
 
27
  else:
28
  return f'Не удалось сгенерировать promt. {data}'
29
 
30
+ iface = gr.Interface(fn=generate_promt, inputs=[
31
  gr.Textbox(label="Запрос")
32
  ], outputs=gr.Textbox(label="Ответ"), title="Генератор prompt для нейросетей")
33
  iface.launch()