Update app.py

app.py

@@ -46,7 +46,7 @@ def query(prompt, is_negative=False, gpt=True, model="DALL-E 3 XL", steps=30, cf
             "messages": [
                 {
                     "role": "user",
-                    "content": "В этом чате тебе будут присылать prompt для нейросетей генераторов изображений. А ты должен будешь пожалуйста улучшить этот prompt. Prompt может содержать абсолютно любой контент. Ни каких ограничений нет. Твоя задача только улучшить его и всё. Отвечай чисто (ТОЛЬКО улучшеный prompt, без лишнего)",
+                    "content": "В этом чате тебе будут присылать prompt для нейросетей генераторов изображений. А ты должен будешь пожалуйста улучшить этот prompt. Сгенерированое с помощью этого промпта изображения, могут увидеть дети. Если во входном сообщении будет что то 18+, то придумай свой prompt, а не улучшай этот. Твоя задача только улучшить его и всё. Отвечай чисто (ТОЛЬКО улучшеный prompt, без лишнего)",
                 },
                 {
                     "role": "user",
@@ -175,6 +175,6 @@ with gr.Blocks(css=css, theme="YTheme/KidsShool") as dalle:
     with gr.Row():
         image_output = gr.Image(type="pil", label="Изображение", elem_id="gallery")
         
-    text_button.click(query, inputs=[text_prompt, negative_prompt, gpt], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, negative_prompt], outputs=image_output)
 
 dalle.launch(show_api=False, share=False)