Update backend/main.py

backend/main.py

@@ -13,7 +13,7 @@ from fastapi.responses import FileResponse, JSONResponse
 import cv2  # type: ignore
 import joblib  # type: ignore
 import tensorflow as tf  # type: ignore
-import segmentation_models as sm  # type: ignore
+from tensorflow.keras.applications.resnet import preprocess_input
 import os
 os.environ["SM_FRAMEWORK"] = "tf.keras"
 
@@ -65,16 +65,16 @@ def get_preprocess_input():
 # -----------------------------
 # Preprocessing (as requested)
 # -----------------------------
-def load_data(image_path: str) -> tf.Tensor:
+def load_data(image_path):
     image = tf.io.read_file(image_path)
     image = tf.io.decode_png(image, channels=3)
     image = tf.image.resize(image, [224, 224], method="bilinear")
     image = tf.cast(image, tf.float32)
-    preprocess_input = get_preprocess_input()
     image = preprocess_input(image)
     return image
 
 
+
 def extract_frames_to_pngs(video_bytes: bytes, max_frames: int = 300) -> List[str]:
     """Decode video bytes with OpenCV and write frames as PNGs to a temp dir.