fix

app.py

@@ -77,10 +77,10 @@ with block:
     def on_click_sam_dencode_btn(inputs):
         print("inferencing")
         image = inputs[raw_image]
-        sam.encode(image)
+        #sam.encode(image)
         generated_masks, _, _ = sam.cond_pred(pts=np.array(inputs[point_coords]), lbls=np.array(inputs[point_labels]))
         annotated = (image, [(generated_masks[i], f'Mask {i}') for i in range(len(generated_masks))])
-        return {masks_annotated_image:annotated, 
+        return {masks_annotated_image: generated_masks, 
                 masks: generated_masks, 
                 cutout_idx: set()}
     sam_encode_btn.click(on_click_sam_encode_btn, components, [masks_annotated_image, masks, cutout_idx], queue=True)