diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/ImageClassificationService.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/ImageClassificationService.java index ddd945ab..a845af9c 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/ImageClassificationService.java +++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/ImageClassificationService.java @@ -12,7 +12,6 @@ import org.springframework.stereotype.Service; import javax.imageio.ImageIO; import java.io.ByteArrayOutputStream; -import java.io.IOException; @Slf4j @Service @@ -37,6 +36,7 @@ public class ImageClassificationService { image.setImageType(ImageType.valueOf(response.getCategory())); } catch (Exception e) { log.error("Could not classify image", e); + image.setImageType(ImageType.OTHER); } log.info("Image classification took: " + (System.currentTimeMillis() - start));