Speed improvements and possible error fix (#2)

- Speed improvements and possible error fix (98e99abd0eba8a6ea507a8c1eb4455023534e829)
- add memory status back (6bf9393a97dc36a1c8927f379ade5fca28be70ce)
- Merge branch 'main' into pr/2 (742c88ec3dd5088470673b25b39c19bd2d615027)


Co-authored-by: Radamés Ajna <[email protected]>

README.md

@@ -2,11 +2,11 @@
 license: openrail
 title: T2I-Adapter
 sdk: gradio
-sdk_version: 3.11
+sdk_version: 3.19.1
 emoji: 😻
 colorFrom: pink
 colorTo: blue
 pinned: false
 python_version: 3.8.16
 app_file: app.py
----
+---

app.py

@@ -1,10 +1,10 @@
 import os
-os.system('pip3 install openmim')
+# os.system('pip3 install openmim')
 os.system('mim install mmcv-full==1.7.0')
-os.system('pip3 install mmpose')
-os.system('pip3 install mmdet')
-os.system('pip3 install gradio==3.19.1')
-os.system('pip3 install psutil')
+# os.system('pip3 install mmpose')
+# os.system('pip3 install mmdet')
+# os.system('pip3 install gradio==3.19.1')
+#os.system('pip3 install psutil')
 
 from demo.model import Model_all
 import gradio as gr
@@ -65,4 +65,4 @@ with gr.Blocks(css='style.css') as demo:
 
 # demo.queue(api_open=False).launch(server_name='0.0.0.0')
 # demo.queue(show_api=False, enable_queue=False).launch(server_name='0.0.0.0')
-demo.queue().launch(debug=True)
+demo.queue().launch(debug=True, server_name='0.0.0.0')

demo/demos.py

@@ -18,6 +18,12 @@ def create_demo_keypose(process):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('T2I-Adapter (Keypose)')
+        with gr.Row():
+            with gr.Column():
+                gr.Textbox(value="Hello Memory")
+            with gr.Column():
+                gr.JSON(get_system_memory, every=1)
+
         with gr.Row():
             with gr.Column():
                 input_img = gr.Image(source='upload', type="numpy")
@@ -43,13 +49,6 @@ def create_demo_sketch(process):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('T2I-Adapter (Sketch)')
-        
-        with gr.Row():
-            with gr.Column():
-                gr.Textbox(value="Hello Memory")
-            with gr.Column():
-                gr.JSON(get_system_memory, every=1)
-
         with gr.Row():
             with gr.Column():
                 input_img = gr.Image(source='upload', type="numpy")
@@ -80,7 +79,7 @@ def create_demo_draw(process):
             with gr.Column():
                 create_button = gr.Button(label="Start", value='Hand-free drawing')
                 input_img = gr.Image(source='upload', type="numpy",tool='sketch')
-                create_button.click(fn=create_map, outputs=[input_img])
+                create_button.click(fn=create_map, outputs=[input_img], queue=False)
                 prompt = gr.Textbox(label="Prompt")
                 neg_prompt = gr.Textbox(label="Negative Prompt",
                 value='ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, bad anatomy, watermark, signature, cut off, low contrast, underexposed, overexposed, bad art, beginner, amateur, distorted face')

requirements.txt

@@ -9,4 +9,10 @@ omegaconf==2.3.0
 taming-transformers-rom1504==0.0.6
 pytorch_lightning==1.8.6
 clip==0.2.0
-kornia==0.6.8
+kornia==0.6.8
+
+
+openmim
+mmpose
+mmdet
+psutil