Make it work with gradio hot-reloading, and reduce startup time by loading less models

app.py

@@ -133,13 +133,13 @@ class ConversationBot:
 
 bot = ConversationBot({'Text2Image': 'cuda:0',
                        'ImageCaptioning': 'cuda:0',
-                       'ImageEditing': 'cuda:0',
-                       'VisualQuestionAnswering': 'cuda:0',
-                       'Image2Canny': 'cpu',
-                       'CannyText2Image': 'cuda:0',
+                    #    'ImageEditing': 'cuda:0',
+                    #    'VisualQuestionAnswering': 'cuda:0',
+                    #    'Image2Canny': 'cpu',
+                    #    'CannyText2Image': 'cuda:0',
                        'InstructPix2Pix': 'cuda:0',
-                       'Image2Depth': 'cpu',
-                       'DepthText2Image': 'cuda:0',
+                    #    'Image2Depth': 'cpu',
+                    #    'DepthText2Image': 'cuda:0',
                        })
 
 with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
@@ -203,4 +203,6 @@ with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
     clear.click(lambda: [], None, chatbot)
     clear.click(lambda: [], None, state)
 
-    demo.queue(concurrency_count=10).launch(server_name="0.0.0.0", server_port=7860)
+    # demo.queue(concurrency_count=10).launch(server_name="0.0.0.0", server_port=7860)
+    if __name__ == "__main__":
+        demo.launch()