MakiPan commited on
Commit
6ab4143
·
1 Parent(s): a0cc202

Update app.py

Browse files

added gradio.radio

Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -86,6 +86,8 @@ def generate_annotation(img, overlap=False, hand_encoding=False):
86
  annotated_image = draw_landmarks_on_image(image.numpy_view(), detection_result, overlap=overlap, hand_encoding=hand_encoding)
87
  return annotated_image
88
 
 
 
89
  args = Namespace(
90
  pretrained_model_name_or_path="runwayml/stable-diffusion-v1-5",
91
  revision="non-ema",
 
86
  annotated_image = draw_landmarks_on_image(image.numpy_view(), detection_result, overlap=overlap, hand_encoding=hand_encoding)
87
  return annotated_image
88
 
89
+ model_type = gr.Radio(["Standard", "Hand Encoding"], label="Model preprocessing", info="We developed two models, one with standard mediapipe landmarks, and one with different (but similar) coloring on palm landmards to distinguish left and right")
90
+
91
  args = Namespace(
92
  pretrained_model_name_or_path="runwayml/stable-diffusion-v1-5",
93
  revision="non-ema",