diff options
Diffstat (limited to 'cmd/bookpipeline/main.go')
-rw-r--r-- | cmd/bookpipeline/main.go | 17 |
1 files changed, 3 insertions, 14 deletions
diff --git a/cmd/bookpipeline/main.go b/cmd/bookpipeline/main.go index 0ffb4b2..425b6a6 100644 --- a/cmd/bookpipeline/main.go +++ b/cmd/bookpipeline/main.go @@ -454,13 +454,7 @@ func ocrPage(msg bookpipeline.Qmsg, conn Pipeliner, process func(chan string, ch errc := make(chan error) msgparts := strings.Split(msg.Body, " ") - bookparts := strings.Split(msgparts[0], "/") - var bookname string - if len(bookparts) > 1 { - bookname = filepath.Dir(msgparts[0]) - } else { - bookname = msgparts[0] - } + bookname := filepath.Dir(msgparts[0]) if len(msgparts) > 1 { process = ocr(msgparts[1]) } @@ -538,13 +532,8 @@ func processBook(msg bookpipeline.Qmsg, conn Pipeliner, process func(chan string errc := make(chan error) msgparts := strings.Split(msg.Body, " ") - bookparts := strings.Split(msgparts[0], "/") - var bookname string - if len(bookparts) > 1 { - bookname = filepath.Dir(msgparts[0]) - } else { - bookname = msgparts[0] - } + bookname := msgparts[0] + var training string if len(msgparts) > 1 { training = msgparts[1] |