Spaces:
Sleeping
Sleeping
Update app.py with transformer embeddings and prediction pipeline
Browse files
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(
|
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()
|