Spaces:
Runtime error
Runtime error
add story_generation_ui
Browse files
app.py
CHANGED
@@ -2,10 +2,14 @@ import gradio as gr
|
|
2 |
|
3 |
from launch.image_generation import image_generation_ui
|
4 |
from launch.model_generation import model_generation_ui
|
|
|
5 |
|
6 |
|
7 |
with gr.Blocks() as demo:
|
8 |
with gr.Group():
|
|
|
|
|
|
|
9 |
with gr.Tab("Generate Image and Remove Background"):
|
10 |
input_image, processed_image = image_generation_ui()
|
11 |
|
|
|
2 |
|
3 |
from launch.image_generation import image_generation_ui
|
4 |
from launch.model_generation import model_generation_ui
|
5 |
+
from launch.story_generator import story_generation_ui
|
6 |
|
7 |
|
8 |
with gr.Blocks() as demo:
|
9 |
with gr.Group():
|
10 |
+
with gr.Tab("Generate Story"):
|
11 |
+
input_image, processed_image = story_generation_ui()
|
12 |
+
|
13 |
with gr.Tab("Generate Image and Remove Background"):
|
14 |
input_image, processed_image = image_generation_ui()
|
15 |
|