Update app.py

app.py

@@ -1,5 +1,5 @@
 from huggingface_hub import hf_hub_download
-import joblib
+import pickle
 import gradio as gr
 import numpy as np
 
@@ -7,7 +7,8 @@ import numpy as np
 model_path = hf_hub_download(repo_id="suryadev1/knn", filename="knn_model.pkl")
 
 # Load the model
-knn = joblib.load(model_path)
+with open(model_path, 'rb') as f:
+    knn = pickle.load(f)
 
 # Define the prediction function
 def predict(input_data):