Update app.py

app.py

@@ -27,7 +27,7 @@ def huff_model_probability(df_distances, df_attractiveness, alpha, beta, df_popu
 
     numerator = (attractiveness_term * distance_term).multiply(adjusted_population, axis=0)
     denominator = numerator.sum(axis=1)
-    probabilities = numerator.div(denominator, axis=0)
+    probabilities = numerator.div(denominator, axis=0).fillna(0)
 
     return probabilities, adjustment_factors
 
@@ -75,8 +75,8 @@ def app_function(input_json):
 
     # Prepare the output
     output = {
-        "probabilities": probabilities.to_json(orient='split'),
-        "adjustment_factors": adjustment_factors.to_json(orient='split')
+        "probabilities": probabilities
+        "adjustment_factors": adjustment_factors
     }
 
     return json.dumps(output)