Update app/services/predictor.py

app/services/predictor.py

@@ -1,30 +1,25 @@
 # app/services/predictor.py
 import numpy as np
 import tensorflow as tf
-from typing import Tuple, Optional
+from typing import Tuple
 
 from app.configs import (
     get_classification_model,
     sigmoid_to_class,
-    CLASSIFIER_URI,
 )
 from app.core.preprocessing import preprocess_image
 from app.core.validation import is_valid_image
 
 
-def classify_image(
-    image_bytes: bytes,
-    model_version: Optional[str] = None,
-) -> Tuple[str, float, str]:
+def classify_image(image_bytes: bytes) -> Tuple[str, float]:
     """
     Classify image using the classification model.
     
     Args:
         image_bytes: Raw image bytes
-        model_version: Optional model version (for A/B testing)
     
     Returns:
-        Tuple of (prediction, confidence, actual_version)
+        Tuple of (prediction, confidence)
     """
     # Validate image
     if not is_valid_image(image_bytes):
@@ -49,10 +44,7 @@ def classify_image(
     # Sigmoid output -> binary classification
     confidence = float(predictions[0][0])
     prediction = sigmoid_to_class(confidence)
-    
-    # Use provided version or default
-    actual_version = model_version or CLASSIFIER_URI
 
     print(f"🔍 CLASSIFICATION: {prediction} (confidence={confidence:.3f})")
 
-    return prediction, confidence, actual_version
+    return prediction, confidence