Update handler.py

handler.py

@@ -12,13 +12,26 @@ class EndpointHandler:
         self.model.eval()
         
     def __call__(self, data):
+        # 디버깅: 입력 데이터 확인
+        print("Received data:", data)
+
         frames = np.array(data['frames'])
         frames = torch.tensor(frames).float()  # Ensure the data is in the correct format
+
+        # 디버깅: 프레임 데이터 확인
+        print("Frames shape:", frames.shape)
         
         # Perform inference
         with torch.no_grad():
             outputs = self.model(frames.unsqueeze(0))  # Add batch dimension
             predictions = torch.softmax(outputs.logits, dim=-1)
+
+            # 디버깅: 예측 결과 확인
+            print("Predictions:", predictions)
+            
             predicted_class = torch.argmax(predictions, dim=-1).item()
+
+            # 디버깅: 예측 클래스 확인
+            print("Predicted class:", predicted_class)
         
         return {"predicted_class": predicted_class, "predictions": predictions.tolist()}