add yandex gpt

app.py

@@ -27,12 +27,30 @@ def get_ingridients(image):
   return ', '.join(ingridients)
 
 
-text_to_text = pipeline("text-generation", model="ai-forever/mGPT")
+from yandex_cloud_ml_sdk import YCloudML
+import os
 
 def get_reciepe(ingridients):
-    question = 'what can i cook of ' + ingridients + '?'
-    answers = text_to_text(question)
-    return answers[0]['generated_text']
+    messages = [
+      {
+        "role": "system",
+        "text": "suggest a dish that can be prepared from the suggested ingredients",
+      },
+      {
+        "role": "user",
+        "text": "spaghetti squash, Granny Smith, cucumber, cuke",
+      },
+    ]
+
+    sdk = YCloudML(
+        folder_id="b1ghdrfjtfkvir55hc0m",
+        auth=os.getenv('YC_APIKEY'),
+    )
+
+    result = (
+        sdk.models.completions("yandexgpt-lite").configure(temperature=0.5).run(messages)
+    )
+    return str(result[0].text)
 
 def get_answer(image):
   ingridients = get_ingridients(image)

requirements.txt

@@ -1,4 +1,4 @@
 transformers
 torch
 Pillow
-
+yandex-cloud-ml-sdk