Абьюзер ChatGPT | 🐍 Python , C++, Freelance

Всем привет. Может, кто хочет попробовать. Генерация картинок, как Midjourney.



1. Основной код:

from diffusers import StableDiffusionPipeline

import torch

from googletrans import Translator



def translate_to_english(text):

    translator = Translator()

    translated = translator.translate(text, src='ru', dest='en')

    return translated.text



def generate_image(prompt, output_path="generated_image.png"):

    model_id = "runwayml/stable-diffusion-v1-5"

    pipe = StableDiffusionPipeline.from_pretrained(model_id)

    if torch.cuda.is_available():

        pipe = pipe.to("cuda")

    print(f"Генерация изображения для запроса: '{prompt}'...")

    image = pipe(prompt).images[0]

    image.save(output_path)

    print(f"Изображение сохранено как {output_path}")



if __name__ == "__main__":

    user_prompt = input("Введите описание изображения: ")

    prompt_in_english = translate_to_english(user_prompt)

    print(f"Переведенный запрос: {prompt_in_english}")

    generate_image(prompt_in_english)

Сделал, чтобы запрос можно было вводить на русском через модуль googletrans (Гугл - транс))))). Для работы кода нужны модули.



2. Модули и команды к их установке:

pip install diffusers transformers accelerate torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

И еще, если вам нужно писать запрос на русском.

pip install googletrans==4.0.0-rc1

3. СКРИПТ ЗАПУСКАЕТСЯ ЧЕРЕЗ ТЕРМИНАЛ!

python image_creator.py

Где "image_creator.py" - название вашего .py файла, куда Вы вставляли код. После - вас попросит ввести описание изображения и выдаст картинку после форматирования. Для этой штуки нужна норм видеокарта, от 1060 и выше. Пробуйте :)