Update handler.py

handler.py

@@ -6,7 +6,7 @@ import logging
 import json
 import traceback
 import os
-from typing import Dict, Any
+from typing import Dict, List, Any
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -22,7 +22,16 @@ class EndpointHandler:
         self.model.eval()
         
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        return {"predicted_class": 1}
+        """
+        data args:
+            inputs (:obj: `str`): base64 encoded video data
+            date (:obj: `str`)
+        Return:
+            A :obj:`list` | `dict`: will be serialized and returned
+        """
+
+        inputs = data.get("inputs")
+        return 1
         
         # inputs = data.get("inputs")
         # if not inputs: