Update handler.py

handler.py

@@ -5,7 +5,8 @@ from preprocessing import read_video
 import logging
 import json
 import traceback
-import base64
+import os
+from fastapi import FastAPI, File, UploadFile
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -20,11 +21,15 @@ class EndpointHandler:
         self.model.classifier = torch.nn.Linear(self.model.classifier.in_features, 48)  # 48 output classes
         self.model.eval()
         
-    def __call__(self, data):
+    def __call__(self, file: UploadFile):
         try:
-            video_base64 = data.get("video_base64")
+            file_path = os.path.join(self.model_dir, file.filename)
+            
+            # 파일을 디스크에 저장
+            with open(file_path, "wb") as buffer:
+                buffer.write(await file.read())
                 
-            processed_frames = read_video(video_base64)
+            processed_frames = read_video(file_path)
     
             return {"processed_frames": processed_frames}