abreza commited on
Commit
4c8ae8c
1 Parent(s): 8df7aef

add story_generation_ui

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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