Update app.py

app.py

@@ -13,14 +13,13 @@ headers = {
     "Authorization": f"Bearer {NEBIUS_API_KEY.strip()}",
     "Content-Type": "application/json"
 }
-
-
     body = {
-        "modelUri": f"gpt://{FOLDER_ID}/yandexgpt/latest",
-        "completionOptions": {"stream": False, "temperature": 0.5, "maxTokens": 1000},
-        "messages": [{"role": "user", "text": prompt}]
+        "folderId": FOLDER_ID,
+        "texts": [input_text],
+        "model": "general",
+        "temperature": 0.6,
+        "maxTokens": "2000"
     }
-
     res = requests.post("https://llm.api.cloud.yandex.net/foundationModels/v1/completion", headers=headers, json=body)
     if res.status_code == 200:
         return res.json()['result']['alternatives'][0]['message']['text']