Update app.py

app.py

@@ -92,6 +92,8 @@ def draw_plot(pred_img, seg):
     ax.tick_params(width=0.0, labelsize=25)
     return fig
 
+seg = None
+
 def on_mouseover(mouse_info):
     x, y = mouse_info["position"]
     
@@ -99,6 +101,7 @@ def on_mouseover(mouse_info):
     highlight_segmentation(x, y)
 
 def highlight_segmentation(x, y):
+    global seg  # 전역 변수로 선언하여 마우스 이벤트에서 사용할 수 있도록 함
     # 세그멘테이션 정보가 None이 아닌 경우에만 출력
     if seg is not None:
         try:
@@ -108,8 +111,9 @@ def highlight_segmentation(x, y):
             print(f"Invalid mouse position: ({x}, {y}), Segmentation value cannot be retrieved.")
     else:
         print("Segmentation information is not available.")
-    
+
 def sepia(input_img):
+    global seg
     input_img = Image.fromarray(input_img)
 
     inputs = feature_extractor(images=input_img, return_tensors="tf")
@@ -119,8 +123,7 @@ def sepia(input_img):
     logits = tf.transpose(logits, [0, 2, 3, 1])
     logits = tf.image.resize(
         logits, input_img.size[::-1]
-    )  # We reverse the shape of `image` because `image.size` returns width and height.
-    global seg 
+    )  # We reverse the shape of `image` because `image.size` returns width and height. 
     seg = tf.math.argmax(logits, axis=-1)[0]
 
     color_seg = np.zeros(