Upload handler.py

handler.py

@@ -3,6 +3,7 @@ import sys
 sys.path.append('./')
 from videollama2 import model_init, mm_infer
 from videollama2.utils import disable_torch_init
+import logging
 
 class EndpointHandler:
     def __init__(self, path: str = ""):
@@ -17,17 +18,17 @@ class EndpointHandler:
         self.model, self.processor, self.tokenizer = model_init(self.model_path)
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        print(f"Received data: {data}")  # Debugging: Print received data
-    
+        logging.info(f"Received data: {data}")  # Debugging: Print received data
+
         modal = data.get("modal", "video")
         modal_path = data.get("modal_path", "")
         instruct = data.get("instruct", "")
-    
-        print(f"Modal: {modal}, Modal Path: {modal_path}, Instruct: {instruct}")  # Debugging: Print extracted values
-    
+
+        logging.info(f"Modal: {modal}, Modal Path: {modal_path}, Instruct: {instruct}")  # Debugging: Print extracted values
+
         if not modal_path or not instruct:
             raise ValueError("Both 'modal_path' and 'instruct' must be provided in the input data.")
-    
+
         # Perform inference
         output = mm_infer(
             self.processor[modal](modal_path), 
@@ -37,7 +38,7 @@ class EndpointHandler:
             do_sample=False, 
             modal=modal
         )
-    
+
         return [{"output": output}]