Marmelator/redaction-service
1
0
Fork 0
Code Issues Pull Requests 14 Actions Packages Projects Releases 622 Wiki Activity

set image type on error

Browse Source
This commit is contained in:
Timo 2021-04-18 11:31:33 +03:00
parent a34d2fb675
commit 42fcea85d3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/ImageClassificationService.java
View File

@ -12,7 +12,6 @@ import org.springframework.stereotype.Service;
import javax.imageio.ImageIO; import javax.imageio.ImageIO;
import java.io.ByteArrayOutputStream; import java.io.ByteArrayOutputStream;
import java.io.IOException;
@Slf4j @Slf4j
@Service @Service
@ -37,6 +36,7 @@ public class ImageClassificationService {
image.setImageType(ImageType.valueOf(response.getCategory())); image.setImageType(ImageType.valueOf(response.getCategory()));
} catch (Exception e) { } catch (Exception e) {
log.error("Could not classify image", e); log.error("Could not classify image", e);
image.setImageType(ImageType.OTHER);
} }
log.info("Image classification took: " + (System.currentTimeMillis() - start)); log.info("Image classification took: " + (System.currentTimeMillis() - start));