Student-gpt / app.py
Rooni's picture
Update app.py
7717b35
raw
history blame
1.2 kB
import gradio as gr
import requests
import json
import os
def gpt3_turbo(image=None, text=None):
# Замените 'YOUR_API_KEY' на ваш API ключ от OpenAI
api_key = os.getenv("API_KEY")
# Формируем данные для отправки запроса
data = {}
if image:
data['image'] = image
if text:
data['text'] = text
# Отправляем запрос на API GPT-3.5-turbo
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {api_key}'
}
endpoint = os.getenv("BASE_URL")
response = requests.post(endpoint, headers=headers, json={"prompt": data, "max_tokens": 50})
if response.status_code == 200:
result = response.json()
return result['choices'][0]['text']
else:
return "Ошибка при получении ответа от модели."
# Создаем интерфейс Gradio
iface = gr.Interface(
fn=gpt3_turbo,
inputs=["text", "text"],
outputs="text",
title="GPT",
description="Помощь ученикам, решение заданий"
)
# Запускаем интерфейс
iface.launch()