abreza's picture
comment image_edition
65a3f6a
import gradio as gr
from launch.image_generation import image_generation_ui
from launch.model_generation import model_generation_ui
from launch.story_generation import story_generation_ui
# from launch.image_edition import image_edition_ui
with gr.Blocks() as demo:
with gr.Group():
with gr.Tab("Generate Story"):
story_generation_ui()
# with gr.Tab("2D Character and Assets"):
# with gr.Tab("Edit Image"):
# image_edition_ui()
with gr.Tab("Generate Image and Remove Background"):
input_image, processed_image = image_generation_ui()
with gr.Tab("3D Model"):
output_model_obj, output_model_glb = model_generation_ui(
processed_image)
demo.launch()