update app.py

app.py

@@ -41,7 +41,8 @@ groundingdino_enable = True
 sam_enable = True
 inpainting_enable = True
 ram_enable = True
-lama_cleaner_enable = True
+
+lama_cleaner_enable = False
 
 kosmos_enable = True
 
@@ -50,11 +51,12 @@ if os.environ.get('IS_MY_DEBUG') is not None:
     inpainting_enable = False
     kosmos_enable = False
 
-try:
-    from lama_cleaner.model_manager import ModelManager
-    from lama_cleaner.schema import Config as lama_Config    
-except Exception as e:
-    lama_cleaner_enable = False
+if lama_cleaner_enable:
+    try:    
+        from lama_cleaner.model_manager import ModelManager
+        from lama_cleaner.schema import Config as lama_Config    
+    except Exception as e:
+        lama_cleaner_enable = False
 
 # segment anything
 from segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator
@@ -68,6 +70,7 @@ from diffusers import StableDiffusionInpaintPipeline
 from huggingface_hub import hf_hub_download
 
 from utils import computer_info
+
 # relate anything
 from ram_utils import iou, sort_and_deduplicate, relation_classes, MLP, show_anns, ram_show_mask
 from ram_train_eval import RamModel, RamPredictor
@@ -84,9 +87,10 @@ if lama_cleaner_enable:
 import ast
 
 if kosmos_enable:
-    # os.system("pip install transformers@git+https://github.com/huggingface/transformers.git@main")
+    os.system("pip install transformers@git+https://github.com/huggingface/transformers.git@main")
     # os.system("pip install transformers==4.32.0")
-    from kosmos_utils import *
+
+from kosmos_utils import *
 
 config_file = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
 ckpt_repo_id = "ShilongLiu/GroundingDINO"

kosmos_utils.py

@@ -6,10 +6,10 @@ import torch
 import torchvision.transforms as torchvision_T
 from PIL import Image
 
-# from transformers import AutoProcessor, AutoModelForVision2Seq
-import subprocess, io, os, sys, time
-sys.path.insert(0, './transformers_4_35_0')
-from transformers_4_35_0 import AutoProcessor, AutoModelForVision2Seq
+from transformers import AutoProcessor, AutoModelForVision2Seq
+# import subprocess, io, os, sys, time
+# sys.path.insert(0, './transformers_4_35_0')
+# from transformers_4_35_0 import AutoProcessor, AutoModelForVision2Seq
 
 import cv2
 import ast