Update app.py

app.py

@@ -100,6 +100,18 @@ def normalbae(img, res):
     return [result]
 
 
+model_dwpose = None
+
+def dwpose(img, res):
+    img = resize_image(HWC3(img), res)
+    global model_dwpose
+    if model_dwpose is None:
+        from annotator.dwpose import DWposeDetector
+        model_dwpose = DWposeDetector()
+    result = model_dwpose(img)
+    return [result]
+
+
 model_openpose = None
 
 
@@ -314,6 +326,18 @@ with block:
             gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
     run_button.click(fn=normalbae, inputs=[input_image, resolution], outputs=[gallery])
 
+    gr.Markdown("<hr>")
+    with gr.Row():
+        gr.Markdown("## DWPose")
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(source='upload', type="numpy")
+            resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)
+            run_button = gr.Button(label="Run")
+        with gr.Column():
+            gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
+    run_button.click(fn=dwpose, inputs=[input_image, resolution], outputs=[gallery])
+
     gr.Markdown("<hr>")
     with gr.Row():
         gr.Markdown("## Openpose")