Text To Image Generator Project

#project text-image generator 
!pip install --upgrade diffusers transformers -q

from pathlib import Path
import tqdm
import torch
import pandas as pd
import numpy as np
from diffusers import StableDiffusionPipeline
from transformers import pipeline, set_seed
import matplotlib.pyplot as plt
import matplotlib.pyplot as plt
import cv2


import torch

class CFG:
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
  seed = 42
  generator = torch.Generator(device=device).manual_seed(seed)
  image_gen_steps = 35
  image_gen_model_id = "stabilityai/stable-diffusion-2"
  image_gen_size = (400, 400)
  image_gen_guidance_scale = 9
  prompt_gen_model_id = "gpt2"
  prompt_dataset_size = 6
  prompt_max_length = 12
  
image_gen_model = StableDiffusionPipeline.from_pretrained(CFG.image_gen_model_id, torch_dtype = torch.float16, revision="fp16", use_auth_token ='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',guidance_scale =9
                                                          )
image_gen_model = image_gen_model.to(CFG.device)

def generate_image(prompt,model):
  image = model(
      prompt,num_inference_steps =CFG.image_gen_steps,
      generator=CFG.generator,
      guidance_scale=CFG.image_gen_guidance_scale
  ).image[0]
  image = image.float16()
  image = image.resize(CFG.image_gen_size)
  return image 


generate_image("two birds", image_gen_model)