Update app.py

app.py

@@ -4,10 +4,13 @@ import numpy as np
 from PIL import Image
 from tensorflow.keras.preprocessing.image import load_img, img_to_array
 from tensorflow.keras.applications.resnet50 import ResNet50, preprocess_input
+import os
+
+BASE_DIR = os.path.dirname(__file__)
 
 # Load the trained KNN model and class names
-knn = joblib.load('knn_model.pkl')  # Replace with the correct path
-class_names = joblib.load('class_names.pkl')  # Replace with the correct path
+knn = joblib.load(os.path.join(BASE_DIR, 'knn_model.pkl'))
+class_names = joblib.load(os.path.join(BASE_DIR, 'class_names.pkl'))
 
 # Function to extract features using ResNet50
 def extract_features(img):