Update app.py

app.py

@@ -112,10 +112,21 @@ def openpose(img, res, hand_and_face):
     result = model_openpose(img, hand_and_face)
     return [result]
 
+model_dwpose = None
 
-model_uniformer = 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_uniformer = None
 
+    
 #def uniformer(img, res):
 #    img = resize_image(HWC3(img), res)
 #    global model_uniformer
@@ -329,6 +340,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=dw_pose, inputs=[input_image, resolution], outputs=[gallery])
+
     gr.Markdown("<hr>")
     with gr.Row():
         gr.Markdown("## Openpose")