abreza's picture
fix issue
80d0821
raw
history blame
598 Bytes
import gradio as gr
from launch.image_generation import image_generation_ui
from launch.model_generation import model_generation_ui
from launch.story_generator import story_generation_ui
with gr.Blocks() as demo:
with gr.Group():
with gr.Tab("Generate Story"):
story_generation_ui()
with gr.Tab("Generate Image and Remove Background"):
input_image, processed_image = image_generation_ui()
with gr.Tab("Generate 3D Model"):
output_model_obj, output_model_glb = model_generation_ui(
processed_image)
demo.launch()