PiKaHa commited on
Commit
a03b8db
1 Parent(s): 4c1817c

Update app.py with transformer embeddings and prediction pipeline

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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()