add markdown to space

app.py

@@ -422,35 +422,6 @@ def get_target_anno(target):
     return pose_img, target_pose, target_cond, target_keypts
 
 
-# def draw_grid(ref):
-#     if ref is None or ref["composite"] is None: # or len(ref["layers"])==0:
-#         return ref
-
-#     # if len(ref["layers"]) == 1:
-#     #     need_draw = True
-#     # # elif ref["composite"].shape[0] != size_memory[0] or ref["composite"].shape[1] != size_memory[1]:
-#     # #     need_draw = True
-#     # else:
-#     #     need_draw = False
-
-#     # size_memory = ref["composite"].shape[0], ref["composite"].shape[1]
-#     # if not need_draw:
-#     #     return size_memory, ref
-
-#     h, w = ref["composite"].shape[:2]
-#     grid_h, grid_w = h // 32, w // 32
-#     # grid = np.zeros((h, w, 4), dtype=np.uint8)
-#     for i in range(1, grid_h):
-#         ref["composite"][i * 32, :, :3] = 255 # 0.5 * ref["composite"][i * 32, :, :3] +
-#     for i in range(1, grid_w):
-#         ref["composite"][:, i * 32, :3] = 255 # 0.5 * ref["composite"][:, i * 32, :3] +
-#     # if len(ref["layers"]) == 1:
-#     #     ref["layers"].append(grid)
-#     # else:
-#     #     ref["layers"][1] = grid
-#     return ref["composite"]
-
-
 def get_mask_inpaint(ref):
     inpaint_mask = np.array(ref["layers"][0])[..., -1]
     inpaint_mask = cv2.resize(
@@ -919,31 +890,6 @@ def set_visible(checkbox, kpts, img_clean, img_pose_right, img_pose_left):
     )
 
 
-# def parse_fix_example(ex_img, ex_masked):
-#     original_img = ex_img
-#     # ex_img = cv2.resize(ex_img, (LENGTH, LENGTH), interpolation=cv2.INTER_AREA)
-#     # ex_masked = cv2.resize(ex_masked, (LENGTH, LENGTH), interpolation=cv2.INTER_AREA)
-#     inpaint_mask = np.all(ex_masked > 250, axis=-1).astype(np.uint8)
-#     layer = np.ones_like(ex_img) * 255
-#     layer = np.concatenate([layer, np.zeros_like(ex_img[..., 0:1])], axis=-1)
-#     layer[inpaint_mask == 1, 3] = 255
-#     ref_value = {
-#         "composite": ex_masked,
-#         "background": ex_img,
-#         "layers": [layer],
-#     }
-#     inpaint_mask = cv2.resize(
-#         inpaint_mask, opts.image_size, interpolation=cv2.INTER_AREA
-#     )
-#     kp_img = visualize_ref(ref_value)
-#     return (
-#         original_img,
-#         gr.update(value=ref_value),
-#         kp_img,
-#         inpaint_mask,
-#     )
-
-
 LENGTH = 480
 
 example_imgs = [
@@ -1012,8 +958,27 @@ custom_css = """
 }
 """
 
+_HEADER_ = '''
+<h2><b>Official Demo</b></h2>
+<h1><b>✋FoundHand: Large-Scale Domain-Specific Learning for Controllable Hand Image Generation</b></h1>
+📝<a href='https://arxiv.org/abs/2412.02690' target='_blank'>Paper</a>
+📢<a href='https://ivl.cs.brown.edu/research/foundhand.html' target='_blank'>Project</a>
+'''
+
+_CITE_ = """
+<h2>Citation</h2>
+```
+    @article{chen2024foundhand,
+    title={FoundHand: Large-Scale Domain-Specific Learning for Controllable Hand Image Generation},
+    author={Chen, Kefan and Min, Chaerin and Zhang, Linguang and Hampali, Shreyas and Keskin, Cem and Sridhar, Srinath},
+    journal={arXiv preprint arXiv:2412.02690},
+    year={2024}
+    }
+```
+"""
 
 with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown(_HEADER_)
     with gr.Tab("Edit Hand Poses"):
         ref_img = gr.State(value=None)
         ref_cond = gr.State(value=None)
@@ -1579,7 +1544,8 @@ with gr.Blocks(css=custom_css) as demo:
                 examples_per_page=20,
             )
 
-
+    gr.Markdown(_CITE_)
+    
 # print("Ready to launch..")
 # _, _, shared_url = demo.queue().launch(
 #     share=True, server_name="0.0.0.0", server_port=7739