Update app.py

app.py

@@ -27,6 +27,7 @@ processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
 global_points = []
 global_point_label = []
 previous_box_points = 0
+draw = None
 
 # Description
 title = "<center><strong><font size='8'> 🍔 Segment food with clicks 🍜</font></strong></center>"
@@ -48,9 +49,13 @@ def read_image(url):
     
     global global_points
     global global_point_label
+    global draw
   
     global_points = []
     global_point_label = []
+    
+    draw = ImageDraw.Draw(image)
+
     return img
 
 # def show_mask(mask, ax, random_color=False):
@@ -180,6 +185,7 @@ def get_points_with_draw(image, label, evt: gr.SelectData):
     global global_points
     global global_point_label
     global previous_box_points
+    global draw
 
     x, y = evt.index[0], evt.index[1]
     point_radius = 15
@@ -190,7 +196,6 @@ def get_points_with_draw(image, label, evt: gr.SelectData):
     print(x, y, label)
     print(previous_box_points)
     
-    draw = ImageDraw.Draw(image)
     if label != 'Bounding Box':
         draw.ellipse([(x - point_radius, y - point_radius), (x + point_radius, y + point_radius)], fill=point_color)
     else: