[camel] 04/10: CAMEL-14808. Fixes for PR review

[acosentino]

This is an automated email from the ASF dual-hosted git repository.

acosentino pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/camel.git

commit 51d860eac76505935f344bdc204a3d4b9dfa79e8
Author: Marat Gubaidullin <marat.gubaidul...@gmail.com>
AuthorDate: Thu Apr 2 09:44:35 2020 -0400

    CAMEL-14808. Fixes for PR review
---
 .../camel/component/djl/model/CustomImageClassificationPredictor.java   | 2 ++
 .../camel/component/djl/model/CustomObjectDetectionPredictor.java       | 2 ++
 .../camel/component/djl/model/ZooImageClassificationPredictor.java      | 2 ++
 .../apache/camel/component/djl/model/ZooObjectDetectionPredictor.java   | 2 ++
 4 files changed, 8 insertions(+)

diff --git 
a/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/CustomImageClassificationPredictor.java
 
b/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/CustomImageClassificationPredictor.java
index 813457b..67106e7 100644
--- 
a/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/CustomImageClassificationPredictor.java
+++ 
b/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/CustomImageClassificationPredictor.java
@@ -63,6 +63,8 @@ public class CustomImageClassificationPredictor extends 
AbstractPredictor {
         } else if (exchange.getIn().getBody() instanceof InputStream) {
             Map<String, Float> result = classify(model, translator, 
exchange.getIn().getBody(InputStream.class));
             exchange.getIn().setBody(result);
+        } else {
+            throw new RuntimeException("Data type is not supported. Body 
should be byte[], InputStream or File");
         }
     }
 
diff --git 
a/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/CustomObjectDetectionPredictor.java
 
b/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/CustomObjectDetectionPredictor.java
index f98b7ca..5cc47be 100644
--- 
a/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/CustomObjectDetectionPredictor.java
+++ 
b/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/CustomObjectDetectionPredictor.java
@@ -61,6 +61,8 @@ public class CustomObjectDetectionPredictor extends 
AbstractPredictor {
         } else if (exchange.getIn().getBody() instanceof InputStream) {
             DetectedObjects result = classify(model, translator, 
exchange.getIn().getBody(InputStream.class));
             exchange.getIn().setBody(result);
+        } else {
+            throw new RuntimeException("Data type is not supported. Body 
should be byte[], InputStream or File");
         }
     }
 
diff --git 
a/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/ZooImageClassificationPredictor.java
 
b/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/ZooImageClassificationPredictor.java
index 4b78585..d78228b 100644
--- 
a/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/ZooImageClassificationPredictor.java
+++ 
b/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/ZooImageClassificationPredictor.java
@@ -68,6 +68,8 @@ public class ZooImageClassificationPredictor extends 
AbstractPredictor {
         } else if (exchange.getIn().getBody() instanceof InputStream) {
             Map<String, Float> result = 
classify(exchange.getIn().getBody(InputStream.class));
             exchange.getIn().setBody(result);
+        } else {
+            throw new RuntimeException("Data type is not supported. Body 
should be byte[], InputStream or File");
         }
     }
 
diff --git 
a/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/ZooObjectDetectionPredictor.java
 
b/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/ZooObjectDetectionPredictor.java
index d90c527..b1e8acf 100644
--- 
a/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/ZooObjectDetectionPredictor.java
+++ 
b/components/camel-djl/src/main/java/org/apache/camel/component/djl/model/ZooObjectDetectionPredictor.java
@@ -66,6 +66,8 @@ public class ZooObjectDetectionPredictor extends 
AbstractPredictor {
         } else if (exchange.getIn().getBody() instanceof InputStream) {
             DetectedObjects result = 
classify(exchange.getIn().getBody(InputStream.class));
             exchange.getIn().setBody(result);
+        } else {
+            throw new RuntimeException("Data type is not supported. Body 
should be byte[], InputStream or File");
         }
     }