Update app.py

app.py

@@ -4,7 +4,6 @@ from fastai.tabular.all import load_learner
 import random
 import torch
 import joblib
-from sklearn.preprocessing import LabelEncoder
 import numpy as np
 
 # Set random seed for reproducibility
@@ -28,6 +27,11 @@ def predict_location(input_data):
     
     # Convert the index to the corresponding location
     location = label_encoder.inverse_transform([pred_idx])[0]
+    
+    # Debugging: Output probabilities and index
+    print(f"Probabilities: {probabilities.numpy()}")
+    print(f"Predicted index: {pred_idx}")
+    
     return location
 
 # Streamlit app