Rename

app.py

@@ -30,17 +30,17 @@ for name in names:
         continue
     subprocess.run(shlex.split(command), cwd='ControlNet/annotator/ckpts/')
 
-from gradio_canny2image import create_demo as create_demo_canny
-from gradio_depth2image import create_demo as create_demo_depth
-from gradio_fake_scribble2image import create_demo as create_demo_fake_scribble
-from gradio_hed2image import create_demo as create_demo_hed
-from gradio_hough2image import create_demo as create_demo_hough
-from gradio_normal2image import create_demo as create_demo_normal
-from gradio_pose2image import create_demo as create_demo_pose
-from gradio_scribble2image import create_demo as create_demo_scribble
-from gradio_scribble2image_interactive import \
+from app_canny import create_demo as create_demo_canny
+from app_depth import create_demo as create_demo_depth
+from app_fake_scribble import create_demo as create_demo_fake_scribble
+from app_hed import create_demo as create_demo_hed
+from app_hough import create_demo as create_demo_hough
+from app_normal import create_demo as create_demo_normal
+from app_pose import create_demo as create_demo_pose
+from app_scribble import create_demo as create_demo_scribble
+from app_scribble_interactive import \
     create_demo as create_demo_scribble_interactive
-from gradio_seg2image import create_demo as create_demo_seg
+from app_seg import create_demo as create_demo_seg
 from model import Model
 
 DESCRIPTION = '''# ControlNet

gradio_canny2image.py → app_canny.py

@@ -81,3 +81,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='canny')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_canny)
+    demo.queue().launch()

gradio_depth2image.py → app_depth.py

@@ -73,3 +73,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='depth')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_depth)
+    demo.queue().launch()

gradio_fake_scribble2image.py → app_fake_scribble.py

@@ -73,3 +73,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='fake_scribble')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_fake_scribble)
+    demo.queue().launch()

gradio_hed2image.py → app_hed.py

@@ -73,3 +73,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='hed')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_hed)
+    demo.queue().launch()

gradio_hough2image.py → app_hough.py

@@ -87,3 +87,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='hough')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_hough)
+    demo.queue().launch()

gradio_normal2image.py → app_normal.py

@@ -80,3 +80,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='normal')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_normal)
+    demo.queue().launch()

gradio_pose2image.py → app_pose.py

@@ -73,3 +73,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='pose')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_pose)
+    demo.queue().launch()

gradio_scribble2image.py → app_scribble.py

@@ -67,3 +67,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='scribble')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_scribble)
+    demo.queue().launch()

gradio_scribble2image_interactive.py → app_scribble_interactive.py

@@ -93,3 +93,10 @@ def create_demo(process, max_images=12, default_num_images=3):
         ]
         run_button.click(fn=process, inputs=inputs, outputs=result)
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_scribble_interactive)
+    demo.queue().launch()

gradio_seg2image.py → app_seg.py

@@ -74,3 +74,10 @@ def create_demo(process, max_images=12, default_num_images=3):
                          outputs=result,
                          api_name='seg')
     return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    model = Model()
+    demo = create_demo(model.process_seg)
+    demo.queue().launch()