From 0fb07d47151c330c5f80c980862707347f46eb56 Mon Sep 17 00:00:00 2001
From: carlosuc3m <100329787@alumnos.uc3m.es>
Date: Wed, 25 May 2022 01:19:16 +0200
Subject: [PATCH] add exception throw when the model fails
---
.../java/tensorflow/v1/Tensorflow1Interface.java | 13 +++++--------
1 file changed, 5 insertions(+), 8 deletions(-)
diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
index 9e605ce..c9ae2ab 100644
--- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
@@ -69,14 +69,11 @@ public class Tensorflow1Interface implements DeepLearningInterface
public void loadModel(String modelFolder, String modelSource) throws LoadModelException {
model = SavedModelBundle.load(modelFolder, "serve");
byte[] byteGraph = model.metaGraphDef();
- try
- {
- sig = MetaGraphDef.parseFrom(byteGraph).getSignatureDefOrThrow("serving_default");
- }
- catch (InvalidProtocolBufferException e)
- {
- System.out.println("Invalid graph");
- }
+ try {
+ sig = MetaGraphDef.parseFrom(byteGraph).getSignatureDefOrThrow("serving_default");
+ } catch (com.google.protobuf.InvalidProtocolBufferException e) {
+ throw new LoadModelException();
+ }
}
@Override
--
GitLab