Examples with AutoMlParams com.google.cloud.documentai.v1beta2.AutoMlParams used on opensource projects

Example 1 with AutoMlParams

use of com.google.cloud.documentai.v1beta2.AutoMlParams in project java-document-ai by googleapis.

the class ParseWithModelBeta method parseWithModel.

public static void parseWithModel(String projectId, String location, String autoMlModel, String gcsUri) throws IOException {
    // the "close" method on the client to safely clean up any remaining background resources.
    try (DocumentUnderstandingServiceClient client = DocumentUnderstandingServiceClient.create()) {
        // Configure the request for processing the PDF
        String parent = String.format("projects/%s/locations/%s", projectId, location);
        AutoMlParams params = AutoMlParams.newBuilder().setModel(autoMlModel).build();
        GcsSource uri = GcsSource.newBuilder().setUri(gcsUri).build();
        // mime_type can be application/pdf, image/tiff,
        // and image/gif, or application/json
        InputConfig config = InputConfig.newBuilder().setGcsSource(uri).setMimeType("application/pdf").build();
        ProcessDocumentRequest request = ProcessDocumentRequest.newBuilder().setParent(parent).setAutomlParams(params).setInputConfig(config).build();
        // Recognizes text entities in the PDF document
        Document response = client.processDocument(request);
        // Process the output
        for (Document.Label label : response.getLabelsList()) {
            System.out.printf("Label detected: %s\n", label.getName());
            System.out.printf("Confidence:  %s\n", label.getConfidence());
        }
    }
}