Update handler.py

handler.py

@@ -101,6 +101,22 @@ class EndpointHandler:
         if 'pixel_values_videos' in inputs_video:
             inputs_video['pixel_values'] = inputs_video.pop('pixel_values_videos')
 
+        if 'pixel_values' in inputs_video:
+            print(f"pixel_values shape: {inputs_video['pixel_values'].shape}")
+        else:
+            print("pixel_values not found in inputs_video")
+
+        if 'input_ids' in inputs_video:
+            print(f"input_ids shape: {inputs_video['input_ids'].shape}")
+        else:
+            print("input_ids not found in inputs_video")
+
+        if 'attention_mask' in inputs_video:
+            print(f"attention_mask shape: {inputs_video['attention_mask'].shape}")
+        else:
+            print("attention_mask not found in inputs_video")
+
+
         # Generate output from the model
         generate_kwargs = {"max_new_tokens": 512, "do_sample": True, "top_p": 0.9}
         output = self.model.generate(**inputs_video, **generate_kwargs)