lev1 commited on
Commit
5ce9bd4
1 Parent(s): c88e7ae

Comming Soon

Browse files
Files changed (3) hide show
  1. app.py +6 -6
  2. app_canny.py +9 -0
  3. app_canny_db.py +11 -0
app.py CHANGED
@@ -3,11 +3,11 @@ import torch
3
 
4
  from model import Model, ModelType
5
 
6
- # from app_canny import create_demo as create_demo_canny
7
  from app_pose import create_demo as create_demo_pose
8
  from app_text_to_video import create_demo as create_demo_text_to_video
9
  from app_pix2pix_video import create_demo as create_demo_pix2pix_video
10
- # from app_canny_db import create_demo as create_demo_canny_db
11
 
12
 
13
  model = Model(device='cuda', dtype=torch.float16)
@@ -47,11 +47,11 @@ with gr.Blocks(css='style.css') as demo:
47
  # pass
48
  create_demo_pose(model)
49
  with gr.Tab('Edge Conditional'):
50
- pass
51
- # create_demo_canny(model)
52
  with gr.Tab('Edge Conditional and Dreambooth Specialized'):
53
- pass
54
- # create_demo_canny_db(model)
55
 
56
  gr.HTML(
57
  """
3
 
4
  from model import Model, ModelType
5
 
6
+ from app_canny import create_demo as create_demo_canny
7
  from app_pose import create_demo as create_demo_pose
8
  from app_text_to_video import create_demo as create_demo_text_to_video
9
  from app_pix2pix_video import create_demo as create_demo_pix2pix_video
10
+ from app_canny_db import create_demo as create_demo_canny_db
11
 
12
 
13
  model = Model(device='cuda', dtype=torch.float16)
47
  # pass
48
  create_demo_pose(model)
49
  with gr.Tab('Edge Conditional'):
50
+ # pass
51
+ create_demo_canny(model)
52
  with gr.Tab('Edge Conditional and Dreambooth Specialized'):
53
+ # pass
54
+ create_demo_canny_db(model)
55
 
56
  gr.HTML(
57
  """
app_canny.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from model import Model
3
+
4
+ def create_demo(model: Model):
5
+
6
+ with gr.Blocks() as demo:
7
+ with gr.Row():
8
+ gr.Markdown('## Text and Canny-Edge Conditional Video Generation (coming soon)')
9
+ return demo
app_canny_db.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from model import Model
3
+ import gradio_utils
4
+
5
+
6
+ def create_demo(model: Model):
7
+
8
+ with gr.Blocks() as demo:
9
+ with gr.Row():
10
+ gr.Markdown('## Text, Canny-Edge and DreamBooth Conditional Video Generation (coming soon)')
11
+ return demo