Removed blip

app.py

@@ -43,8 +43,8 @@ from GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases
 # segment anything
 from segment_anything import build_sam_vit_l, SamPredictor 
 
-# BLIP
-from transformers import BlipProcessor, BlipForConditionalGeneration
+# # BLIP
+# from transformers import BlipProcessor, BlipForConditionalGeneration
 
 # Constants
 CONFIG_FILE = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
@@ -55,9 +55,7 @@ OUTPUT_DIR = "outputs"
 # Global variables for model caching
 _models = {
     'groundingdino': None,
-    'sam_predictor': None,
-    'blip_processor': None,
-    'blip_model': None
+    'sam_predictor': None
 }
 
 # Enable GPU if available with proper error handling
@@ -107,16 +105,16 @@ class ModelManager:
                 
                 logger.info(f"SAM-HQ model loaded in {time.time() - start_time:.2f} seconds")
                 
-            elif model_name == 'blip' and (_models['blip_processor'] is None or _models['blip_model'] is None):
-                logger.info("Loading BLIP model...")
-                start_time = time.time()
+            # elif model_name == 'blip' and (_models['blip_processor'] is None or _models['blip_model'] is None):
+            #     logger.info("Loading BLIP model...")
+            #     start_time = time.time()
                 
-                _models['blip_processor'] = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
-                _models['blip_model'] = BlipForConditionalGeneration.from_pretrained(
-                    "Salesforce/blip-image-captioning-large", torch_dtype=torch.float16
-                ).to(device)
+            #     _models['blip_processor'] = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
+            #     _models['blip_model'] = BlipForConditionalGeneration.from_pretrained(
+            #         "Salesforce/blip-image-captioning-large", torch_dtype=torch.float16
+            #     ).to(device)
                 
-                logger.info(f"BLIP model loaded in {time.time() - start_time:.2f} seconds")
+            #     logger.info(f"BLIP model loaded in {time.time() - start_time:.2f} seconds")
                 
         except Exception as e:
             logger.error(f"Error loading {model_name} model: {e}")