salomonsky commited on
Commit
b6ca7e1
·
1 Parent(s): 93fc9d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +29 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import torch
3
- from scipy.ndimage import gaussian_filter
4
  from model import Model, ModelType
5
  from app_canny import create_demo as create_demo_canny
6
  from app_pose import create_demo as create_demo_pose
@@ -19,6 +19,34 @@ parser.add_argument('--public_access', action='store_true',
19
  help="if enabled, the app can be access from a public url", default=False)
20
  args = parser.parse_args()
21
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  if on_huggingspace:
23
  demo.queue(max_size=20)
24
  demo.launch(debug=True)
 
1
  import gradio as gr
2
  import torch
3
+
4
  from model import Model, ModelType
5
  from app_canny import create_demo as create_demo_canny
6
  from app_pose import create_demo as create_demo_pose
 
19
  help="if enabled, the app can be access from a public url", default=False)
20
  args = parser.parse_args()
21
 
22
+
23
+ with gr.Blocks(css='style.css') as demo:
24
+ gr.HTML(
25
+ """
26
+ """)
27
+
28
+ if on_huggingspace:
29
+ gr.HTML("""
30
+ """)
31
+
32
+ with gr.Tab('Zero-Shot Text2Video'):
33
+ create_demo_text_to_video(model)
34
+ with gr.Tab('Video Instruct Pix2Pix'):
35
+ create_demo_pix2pix_video(model)
36
+ with gr.Tab('Pose Conditional'):
37
+ create_demo_pose(model)
38
+ with gr.Tab('Edge Conditional'):
39
+ create_demo_canny(model)
40
+ with gr.Tab('Edge Conditional and Dreambooth Specialized'):
41
+ create_demo_canny_db(model)
42
+ with gr.Tab('Depth Conditional'):
43
+ create_demo_depth(model)
44
+ '''
45
+ '''
46
+ gr.HTML(
47
+ """
48
+ """)
49
+
50
  if on_huggingspace:
51
  demo.queue(max_size=20)
52
  demo.launch(debug=True)