Student-gpt / app.py
Rooni's picture
Update app.py
2cbe2b0
raw
history blame
1.8 kB
# app.py
import os
import gradio as gr
import openai
# Установка ключа API OpenAI
openai.api_key = os.getenv("API_KEY")
# Функция для генерации решения с использованием GPT-3.5-turbo
def generate_solution(prompt, image):
# Ваша логика обработки изображения и текста перед отправкой модели
# Здесь должен быть ваш код для обработки изображения и текста
# Пример: конкатенация текста и описания изображения
input_text = f"{prompt} {image}"
# Запрос к модели GPT-3.5-turbo
response = openai.Completion.create(
engine="text-davinci-003", # Выбор движка GPT
prompt=input_text,
max_tokens=150, # Максимальное количество токенов в ответе
n=1, # Количество ответов, которые вы хотите получить
)
# Извлечение сгенерированного текста из ответа модели
generated_text = response["choices"][0]["text"]
return generated_text
# Определение интерфейса Gradio
iface = gr.Interface(
fn=generate_solution,
inputs=[
gr.Textbox("Введите описание задачи", lines=5), # Используем параметр lines для многострочного текстового поля
gr.Image(type="pil", label="Загрузите изображение"),
],
outputs=gr.Textbox("Решение задачи"),
live=True,
theme="huggingface",
)
# Запуск приложения Gradio
iface.launch(share=True)