update app.py

app.py

@@ -37,8 +37,14 @@ from GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases
 import cv2
 import numpy as np
 import matplotlib.pyplot as plt
-from lama_cleaner.model_manager import ModelManager
-from lama_cleaner.schema import Config as lama_Config
+
+lama_cleaner_enable = False
+try:
+    from lama_cleaner.model_manager import ModelManager
+    from lama_cleaner.schema import Config as lama_Config
+    lama_cleaner_enable = True
+except Exception as e:
+    pass
 
 # segment anything
 from segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator
@@ -56,11 +62,13 @@ from utils import computer_info
 from ram_utils import iou, sort_and_deduplicate, relation_classes, MLP, show_anns, ram_show_mask
 from ram_train_eval import RamModel,RamPredictor
 from mmengine.config import Config as mmengine_Config
-from lama_cleaner.helper import (
-    load_img,
-    numpy_to_bytes,
-    resize_max_size,
-)
+
+if lama_cleaner_enable:
+    from lama_cleaner.helper import (
+        load_img,
+        numpy_to_bytes,
+        resize_max_size,
+    )
 
 # from transformers import AutoProcessor, AutoModelForVision2Seq
 import ast
@@ -787,7 +795,9 @@ def get_model_device(module):
 ram_enable = True
 kosmos_enable = True
 
-task_types = ["detection", "segment", "inpainting", "remove"]
+task_types = ["detection", "segment", "inpainting"]
+if lama_cleaner_enable:
+    task_types.append("remove")
 if ram_enable:
     task_types.append("relate anything")
 if kosmos_enable:
@@ -811,7 +821,10 @@ if __name__ == "__main__":
         load_sam_model()
 
     load_sd_model()
-    load_lama_cleaner_model()
+
+    if lama_cleaner_enable:
+        load_lama_cleaner_model()
+
     if ram_enable:
         load_ram_model()
     
@@ -907,7 +920,8 @@ if __name__ == "__main__":
         DESCRIPTION = f'### This demo from [Grounded-Segment-Anything](https://github.com/IDEA-Research/Grounded-Segment-Anything). <br>'
         if ram_enable:
             DESCRIPTION += f'RAM from [RelateAnything](https://github.com/Luodian/RelateAnything). <br>'
-        DESCRIPTION += f'Remove(cleaner) from [lama-cleaner](https://github.com/Sanster/lama-cleaner). <br>'
+        if lama_cleaner_enable:
+            DESCRIPTION += f'Remove(cleaner) from [lama-cleaner](https://github.com/Sanster/lama-cleaner). <br>'
         if kosmos_enable:
             DESCRIPTION += f'Kosmos-2 from [Kosmos-2](https://github.com/microsoft/unilm/tree/master/kosmos-2). <br>'
         DESCRIPTION += f'Thanks for their excellent work.'

requirements.txt

@@ -26,6 +26,6 @@ safetensors
 pynvml
 sentencepiece
 
-lama-cleaner==1.2.3
+# lama-cleaner==1.2.3
 openmim==0.1.5
 mmcv==2.0.0