Spaces:
Running
Running
File size: 858 Bytes
8d9a1a3 c5dcd31 8d9a1a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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):
# return I2I(image, prompt)
def Image2Image(
input_image_editor: dict,
input_text: str,
seed_slicer: int,
randomize_seed_checkbox: bool,
strength_slider: float,
num_inference_steps_slider: int,
):return I2I(input_image_editor,input_text,seed_slicer,randomize_seed_checkbox,strength_slider,num_inference_steps_slider)
def Image2Image_2(prompt,image,size,num_inference_steps=30):
return I2I_2(image, prompt,size,num_inference_steps) |