Update handler.py

handler.py

@@ -27,13 +27,11 @@ class EndpointHandler:
         self.model = AutoModelForCausalLM.from_pretrained(
             self.model_name,
             trust_remote_code=True,
-            revision='refs/pr/6'
         ).to(self.device)
         
         self.processor = AutoProcessor.from_pretrained(
             self.model_name,
             trust_remote_code=True,
-            revision='refs/pr/6'
         )
         
         if torch.cuda.is_available():
@@ -61,10 +59,10 @@ class EndpointHandler:
                 # If inputs is not a dict, assume it's the image path
                 image_path = inputs
                 text_input = "What is in this image?"
-            
+            print("[INFO]",image_path,text_input)
             # Process image
             image = self.process_image(image_path) if image_path else None
-            
+            print("[INFO]",image)
             # Prepare inputs for the model
             model_inputs = self.processor(
                 images=image if image else None,