Update app.py

app.py

@@ -8,7 +8,7 @@ from PIL import ImageDraw
 from torchvision.transforms import ToTensor
 
 from utils.tools import format_results, point_prompt
-from utils.tools_gradio import fast_process
+from utils.tools_gradio import fast_process, fast_show_mask
 
 # Most of our demo code is from [FastSAM Demo](https://huggingface.co/spaces/An-619/FastSAM). Thanks for AN-619.
 
@@ -135,7 +135,8 @@ def segment_with_boxs(
     annotations = results[0]["segmentation"]
     annotations = np.array([annotations])
     print(scaled_points.shape)
-    fig = fast_process(
+    # change this
+    fig = fast_show_mask(
         annotations=annotations,
         image=image,
         device=device,
@@ -214,7 +215,8 @@ def segment_with_points(
     )
     annotations = np.array([annotations])
 
-    fig = fast_process(
+    # change this
+    fig = fast_show_mask(
         annotations=annotations,
         image=image,
         device=device,