import torch
from options.Banner_Model.Text2Banner import T2I
from options.Banner_Model.Image2Image import I2I
from options.Banner_Model.Image2Image_2 import I2I_2

def TextImage(prompt, width=1024, height=1024, guidance_scale=3.5,
              num_inference_steps=28):
    img = T2I(prompt, width, height, guidance_scale, num_inference_steps)
    return img

def Image2Image(prompt,image, width=1024, height=1024, guidance_scale=3.5,num_inference_steps=28,strength=0.99):
    return I2I(prompt, image, width, height, guidance_scale, num_inference_steps, strength)

def Image2Image_2(prompt,image,size,num_inference_steps=30):
    return I2I_2(image, prompt,size,num_inference_steps)