diff --git a/automl/cloud-client/src/main/java/com/example/automl/LanguageEntityExtractionPredict.java b/automl/cloud-client/src/main/java/com/example/automl/LanguageEntityExtractionPredict.java index 1294c609f73..4f98f3741aa 100644 --- a/automl/cloud-client/src/main/java/com/example/automl/LanguageEntityExtractionPredict.java +++ b/automl/cloud-client/src/main/java/com/example/automl/LanguageEntityExtractionPredict.java @@ -44,8 +44,14 @@ static void predict(String projectId, String modelId, String content) throws IOE try (PredictionServiceClient client = PredictionServiceClient.create()) { // Get the full path of the model. ModelName name = ModelName.of(projectId, "us-central1", modelId); + + // For available mime types, see: + // https://cloud.google.com/automl/docs/reference/rest/v1/projects.locations.models/predict#textsnippet TextSnippet textSnippet = - TextSnippet.newBuilder().setContent(content).setMimeType("text/plain").build(); + TextSnippet.newBuilder() + .setContent(content) + .setMimeType("text/plain") // Types: text/plain, text/html + .build(); ExamplePayload payload = ExamplePayload.newBuilder().setTextSnippet(textSnippet).build(); PredictRequest predictRequest = PredictRequest.newBuilder().setName(name.toString()).setPayload(payload).build(); diff --git a/automl/cloud-client/src/main/java/com/example/automl/LanguageSentimentAnalysisPredict.java b/automl/cloud-client/src/main/java/com/example/automl/LanguageSentimentAnalysisPredict.java index c07af6fb5c5..65945c26023 100644 --- a/automl/cloud-client/src/main/java/com/example/automl/LanguageSentimentAnalysisPredict.java +++ b/automl/cloud-client/src/main/java/com/example/automl/LanguageSentimentAnalysisPredict.java @@ -43,6 +43,9 @@ static void predict(String projectId, String modelId, String content) throws IOE try (PredictionServiceClient client = PredictionServiceClient.create()) { // Get the full path of the model. ModelName name = ModelName.of(projectId, "us-central1", modelId); + + // For available mime types, see: + // https://cloud.google.com/automl/docs/reference/rest/v1/projects.locations.models/predict#textsnippet TextSnippet textSnippet = TextSnippet.newBuilder() .setContent(content) diff --git a/automl/cloud-client/src/main/java/com/example/automl/LanguageTextClassificationPredict.java b/automl/cloud-client/src/main/java/com/example/automl/LanguageTextClassificationPredict.java index 64cd76e9287..e2da94aa121 100644 --- a/automl/cloud-client/src/main/java/com/example/automl/LanguageTextClassificationPredict.java +++ b/automl/cloud-client/src/main/java/com/example/automl/LanguageTextClassificationPredict.java @@ -43,6 +43,9 @@ static void predict(String projectId, String modelId, String content) throws IOE try (PredictionServiceClient client = PredictionServiceClient.create()) { // Get the full path of the model. ModelName name = ModelName.of(projectId, "us-central1", modelId); + + // For available mime types, see: + // https://cloud.google.com/automl/docs/reference/rest/v1/projects.locations.models/predict#textsnippet TextSnippet textSnippet = TextSnippet.newBuilder() .setContent(content)