update

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from demo import SegAutoMaskPredictor
+from demo import automask_image_app, automask_video_app
 
 
 def image_app():
@@ -47,7 +47,7 @@ def image_app():
                 output_image = gr.Image()
 
         seg_automask_image_predict.click(
-            fn=SegAutoMaskPredictor().image_predict,
+            fn=automask_image_app,
             inputs=[
                 seg_automask_image_file,
                 seg_automask_image_model_type,
@@ -103,7 +103,7 @@ def video_app():
                 output_video = gr.Video()
 
         seg_automask_video_predict.click(
-            fn=SegAutoMaskPredictor().video_predict,
+            fn=automask_video_app,
             inputs=[
                 seg_automask_video_file,
                 seg_automask_video_model_type,

demo.py

@@ -3,7 +3,7 @@ from metaseg import SegAutoMaskPredictor, SegManualMaskPredictor, SahiAutoSegmen
 # For image
 
 
-def image_app(image_path, model_type, points_per_side, points_per_batch, min_area):
+def automask_image_app(image_path, model_type, points_per_side, points_per_batch, min_area):
     SegAutoMaskPredictor().image_predict(
         source=image_path,
         model_type=model_type,  # vit_l, vit_h, vit_b
@@ -20,7 +20,7 @@ def image_app(image_path, model_type, points_per_side, points_per_batch, min_are
 # For video
 
 
-def video_app(video_path, model_type, points_per_side, points_per_batch, min_area):
+def automask_video_app(video_path, model_type, points_per_side, points_per_batch, min_area):
     SegAutoMaskPredictor().video_predict(
         source=video_path,
         model_type=model_type,  # vit_l, vit_h, vit_b