update requirements.txt

models/blip2_model.py

@@ -11,8 +11,8 @@ class ImageCaptioning:
         self.processor, self.model = self.initialize_model()
 
     def initialize_model(self):
-        # device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.device = "cpu" # for low gpu memory devices
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        # self.device = "cpu" # for low gpu memory devices
         if self.device == 'cpu':
             self.data_type = torch.float32
         else:

models/segment_models/semantic_segment_anything_model.py

@@ -19,8 +19,8 @@ class SemanticSegment():
         self.model_init()
 
     def model_init(self):
-        # self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.device = 'cpu'
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        # self.device = 'cpu'
         self.init_clip()
         self.init_oneformer_ade20k()
         self.init_oneformer_coco()

requirements.txt

@@ -16,4 +16,6 @@ spacy
 en_core_web_sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.0.0/en_core_web_sm-3.0.0.tar.gz
 lvis
 scipy
-boto3
+boto3
+jsonschema
+entrypoints