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

Update app.py with transformer embeddings and prediction pipeline

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -73,7 +73,8 @@ def predict_with_gpflow(model, X):
73
  X_tensor = tf.convert_to_tensor(X, dtype=tf.float64)
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
 
73
  X_tensor = tf.convert_to_tensor(X, dtype=tf.float64)
74
  print (X_tensor.shape)
75
  # Get predictions
76
+ #predict_fn = model.predict_f_compiled
77
+ predict_fn = model.signatures["serving_default"]
78
  mean, variance = predict_fn(Xnew=X_tensor)
79
 
80
  # Return mean and variance as numpy arrays