Skip to content

Commit

Permalink
Update gemma2PredictGpu.js
Browse files Browse the repository at this point in the history
Move the region tag to show to documentation users how to call the function.
  • Loading branch information
rsamborski authored and iennae committed Oct 15, 2024
1 parent 35bc528 commit e587ab9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ai-platform/snippets/gemma2PredictGpu.js
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@

'use strict';

// [START generativeaionvertexai_gemma2_predict_gpu]
async function gemma2PredictGpu(predictionServiceClient) {
// [START generativeaionvertexai_gemma2_predict_gpu]
// Imports the Google Cloud Prediction Service Client library
const {
// TODO(developer): Uncomment PredictionServiceClient before running the sample.
Expand Down Expand Up @@ -62,7 +62,6 @@ async function gemma2PredictGpu(predictionServiceClient) {
const text = predictions[0].stringValue;

console.log('Predictions:', text);
// [END generativeaionvertexai_gemma2_predict_gpu]
return text;
}

Expand All @@ -73,3 +72,4 @@ module.exports = gemma2PredictGpu;
// console.error(err.message);
// process.exitCode = 1;
// });
// [END generativeaionvertexai_gemma2_predict_gpu]

0 comments on commit e587ab9

Please sign in to comment.