Update app.py

app.py

@@ -35,10 +35,10 @@ def predict_score(lat, lon):
 
     # Get model output
     with torch.no_grad():
-        outputs = model(inputs).numpy().flatten()
+        mu_pred, alpha = model(inputs)
 
     # Unpack into respective values
-    mu_pred, alpha = outputs
+    mu_pred = mu_pred.numpy().flatten()
 
     score = (1 * np.abs(mu_pred + 0.1)) * 100