Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,14 +3,14 @@ import requests
|
|
3 |
import json
|
4 |
import os
|
5 |
|
6 |
-
def
|
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=
|
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()
|