Update app.py

app.py

@@ -1,19 +1,10 @@
-import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
-import torch
+import openai
 
-# Загрузка модели (можно заменить на ruGPT-3 или другую)
-model_name = "sberbank-ai/rugpt3small_based_on_gpt2"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+openai.api_key = "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"  # Вставьте ваш API-ключ
 
-# Функция для генерации ответа
-def generate_text(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt")
-    with torch.no_grad():
-        outputs = model.generate(**inputs, max_new_tokens=50)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+response = openai.ChatCompletion.create(
+    model="gpt-3.5-turbo",
+    messages=[{"role": "user", "content": "Привет! Что ты умеешь?"}]
+)
 
-# Интерфейс Gradio
-demo = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="Тест модели")
-demo.launch()
+print(response['choices'][0]['message']['content'])