Spaces:
Running
Running
File size: 667 Bytes
8d9a1a3 bc8d897 4f62723 8d9a1a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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, height, guidance_scale,num_inference_steps,strength
):
return I2I(image, prompt, 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) |