Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import json
 import os
 
-def generate(description):
+def generate(description, model, max_tokens):
     headers = {
         'Content-Type': 'application/json',
         'Authorization': f'Bearer {os.getenv("API_KEY")}'
@@ -11,8 +11,8 @@ def generate(description):
 
     payload = {
         'messages': [{'role': 'system', 'content': f'{description}'}],
-        'max_tokens': 10000,
-        'model': os.getenv("MODEL")
+        'max_tokens': max_tokens,
+        'model': model
     }
 
     response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
@@ -28,6 +28,8 @@ def generate(description):
         return f'Не удалось сгенерировать текст. {data}'
 
 iface = gr.Interface(fn=generate, inputs=[
-    gr.Textbox(label="Запрос")
+    gr.Textbox(label="Запрос"),
+    gr.Radio(show_label=True, label="Модель", interactive=True, choices=["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4"], value="gpt-3.5-turbo"),
+    gr.Slider(show_label=True, label="Максимум токенов", minimum=100, maximum=15000, value=10000, step=1)
 ], outputs=gr.Textbox(label="Ответ"), title="GPT")
 iface.launch()