Update app.py

app.py

@@ -50,7 +50,7 @@ def classify_image(uploaded_image):
     input_image = data_transforms(image).unsqueeze(0).to(device)
 
     # Anomaly detection
-    if is_anomaly(clf, feature_extractor):
+    if is_anomaly(clf, feature_extractor, input_image):
         return "Anomaly detected. Image will not be classified.", None
 
     # Classification