Dall-E: генерация изображений на Python

Этот урок описывает работу с моделью Dall-E для генерации изображений на основе текстового описания. Рассмотрен практический пример создания функции на Python, взаимодействующей с API Dall-E и выводящей сгенерированную ссылку на изображение.

Подготовка к работе

Перед началом работы, посетите сайт itproger.com. Там вы найдете код, домашнее задание и другую полезную информацию. Ссылка на данный урок будет размещена в описании.

Работа с API Dall-E

В отличие от работы с моделью GPT, здесь используется Dall-E для генерации изображений. Результаты не всегда идеальны, поскольку изображения создаются искусственным интеллектом с нуля.

Реализация на Python

Пример не использует библиотеку OpenAI. Вместо этого, осуществляется прямое обращение к URL-адресу API Dall-E с передачей параметров через POST-запрос. Ответ API содержит ссылку на сгенерированное изображение, которую можно вывести в консоль или использовать в других приложениях (например, в Telegram-боте).

Функция generate

Функция generate принимает текстовое сообщение (prompt) для генерации изображения:

import requests

def generate(message):
    url = "YOUR_DALL_E_API_URL"  # Замените на ваш URL
    headers = {
        "Authorization": f"Bearer YOUR_API_KEY",  # Замените на ваш API ключ
        "Content-Type": "application/json"
    }
    response = requests.post(url, json={"prompt": message}, headers=headers)
    if response.status_code == 200:
        result = response.json()
        print(result)  # Вывод результата в консоль
    else:
        print(f"Error: {response.status_code} - {response.text}")

Заметьте: Необходимо заменить "YOUR_DALL_E_API_URL" на ваш URL API и "YOUR_API_KEY" на ваш API ключ. Обратите внимание на изменение заголовка Authorization.

Установка и импорт библиотеки requests

Для работы с HTTP-запросами необходима библиотека requests. Установите её с помощью pip:

pip install requests

Затем импортируйте библиотеку в скрипт: import requests

Генерация и вывод изображения

Переменная text содержит текстовое описание для генерации изображения. Функция generate вызывается с этим описанием.

text = "Bugatti and Volvo"
generate(text)

В консоли будет выведен JSON-ответ, содержащий ссылку на сгенерированное изображение.

Анализ результатов

Качество изображения зависит от детализации описания. Более подробное описание даст более точные результаты. В примере описание («Bugatti and Volvo») общее, поэтому результат может отличаться от ожидаемого.

Dall-E — мощный инструмент для генерации изображений на основе текста. Простой Python-скрипт позволяет взаимодействовать с API Dall-E и создавать уникальные изображения. Экспериментируйте с различными описаниями для достижения наилучших результатов. Чем подробнее описание, тем лучше результат.