Update app.py

app.py

@@ -4,13 +4,13 @@ import joblib
 import numpy as np
 
 from preprocess_cnn import preprocess_for_cnn
-from preprocess_quality import extract_quality_features
+from preprocess_quality import GeometricFeatureExtractor
 
 cnn_model = tf.keras.models.load_model("custom_cnn_shape.keras")
 xgb_quality = joblib.load("xgb_quality.pkl")
 scaler = joblib.load("scaler.pkl")
 le_quality = joblib.load("le_quality.pkl")
-
+feature_extractor = GeometricFeatureExtractor()
 SHAPES = ['Triangle', 'Square', 'Circle', 'Rectangle']
 
 def predict(image):
@@ -23,7 +23,7 @@ def predict(image):
     response["shape"] = SHAPES[shape_idx]
     response["shape_confidence"] = float(shape_probs[shape_idx])
 
-    features = extract_quality_features(image)
+    features = feature_extractor.extract_features(image)
 
     if features is None:
         response["quality"] = "unknown"