sergiopaniego HF Staff commited on
Commit
bf13da5
·
1 Parent(s): 1f2ca3d
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -73,6 +73,9 @@ def process_inputs(prompts):
73
  elif input_points:
74
  img1_b64 = show_all_annotations_on_image_base64(user_image, sam_masks[0][0], sam_scores[:, 0, :], None, input_points[0], model_name='SAM')
75
  img2_b64 = show_all_annotations_on_image_base64(user_image, sam_hq_masks[0][0], sam_hq_scores[:, 0, :], None, input_points[0], model_name='SAM_HQ')
 
 
 
76
 
77
  print('user_image', user_image)
78
  print("img1_b64", img1_b64)
 
73
  elif input_points:
74
  img1_b64 = show_all_annotations_on_image_base64(user_image, sam_masks[0][0], sam_scores[:, 0, :], None, input_points[0], model_name='SAM')
75
  img2_b64 = show_all_annotations_on_image_base64(user_image, sam_hq_masks[0][0], sam_hq_scores[:, 0, :], None, input_points[0], model_name='SAM_HQ')
76
+ else:
77
+ img1_b64 = show_all_annotations_on_image_base64(user_image, None, None, None, None, model_name='SAM')
78
+ img2_b64 = show_all_annotations_on_image_base64(user_image, None, None, None, None, model_name='SAM_HQ')
79
 
80
  print('user_image', user_image)
81
  print("img1_b64", img1_b64)