Update app.py

app.py

@@ -7,8 +7,6 @@ import pandas as pd
 
 from utils import compute_features
 
-dataset = pd.read_csv("cleaned_df_noindex.csv")
-df_amenities = pd.read_csv("df_amenities.csv")
 
 class NegBinomialModel(nn.Module):
     def __init__(self, in_features):
@@ -32,14 +30,16 @@ def negbinom_loss(y, mu, alpha):
     )
     return -torch.mean(log_prob)
     
-model = NegBinomialModel(len(dataset.columns)-2)
+model = NegBinomialModel(17)
 model.load_state_dict(torch.load("model_weights.pt"))
 model.eval()
 
-# ======== Prediction Function ========
 def predict_score(lat, lon):
     # Convert input to tensor
-    inputs = torch.tensor([[lat, lon]], dtype=torch.float32)
+    # inputs = torch.tensor([[lat, lon]], dtype=torch.float32)
+    inputs = compute_features((lat,lon))
+
+    inputs = torch.tensor([lat,lon] + list(inputs.values))
 
     # Get model output
     with torch.no_grad():