PiKaHa commited on
Commit
a5b4eef
·
1 Parent(s): 3ded8ec

Update app.py with transformer embeddings and prediction pipeline

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ def predict_with_gpflow(model, X):
74
  print (X_tensor.shape)
75
  # Get predictions
76
  predict_fn = model.predict_f_compiled
77
- mean, variance = predict_fn(X_new=X_tensor)
78
 
79
  # Return mean and variance as numpy arrays
80
  return mean.numpy().flatten(), variance.numpy().flatten()
 
74
  print (X_tensor.shape)
75
  # Get predictions
76
  predict_fn = model.predict_f_compiled
77
+ mean, variance = predict_fn(Xnew=X_tensor)
78
 
79
  # Return mean and variance as numpy arrays
80
  return mean.numpy().flatten(), variance.numpy().flatten()