logging

handler.py

@@ -8,19 +8,19 @@ from cv2 import imencode
 from base64 import b64encode
 import time
 
-logging.set_verbosity_info()
+# logging.set_verbosity_info()
 
 class EndpointHandler():
   def __init__(self, path=""):
     self.device = "cuda" if torch.cuda.is_available() else "cpu"
     self.logger = logging.get_logger("transformers")
-    self.logger.info(f"Using device: {self.device}")
+    self.logger.warn(f"Using device: {self.device}")
     self.generator = pipeline("mask-generation", model="facebook/sam-vit-large", device=self.device)
 
   def __call__(self, data):
     start = time.time()
     inputs = data.pop("inputs", data)
-    self.logger.info(f"got request for {inputs}")
+    self.logger.warn(f"got request for {inputs}")
     raw_image = Image.open(requests.get(inputs, stream=True).raw).convert("RGB")
     
     with torch.no_grad():