Spaces:
Sleeping
Sleeping
Update app.py with transformer embeddings and prediction pipeline
Browse files
app.py
CHANGED
|
@@ -75,6 +75,7 @@ def predict_with_gpflow(model, X):
|
|
| 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 |
mean = result["output_0"].numpy() # Adjust output key names if needed
|
| 80 |
variance = result["output_1"].numpy()
|
|
|
|
| 75 |
# Get predictions
|
| 76 |
#predict_fn = model.predict_f_compiled
|
| 77 |
predict_fn = model.signatures["serving_default"]
|
| 78 |
+
result = predict_fn(Xnew=X_tensor) # Pass Xnew explicitly
|
| 79 |
#mean, variance = predict_fn(Xnew=X_tensor)
|
| 80 |
mean = result["output_0"].numpy() # Adjust output key names if needed
|
| 81 |
variance = result["output_1"].numpy()
|