ocr class

work-design · Dec 16, 2024 · 4376d60 · 4376d60
1 parent 972f057
commit 4376d60
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/app/apis/ali_ocr.rb b/app/apis/ali_ocr.rb
@@ -35,6 +35,26 @@ def ocr_template(file, task_id:)
     end
   end
 
+  def ocr_class(file, class_id:)
+    body = {
+      action: 'PredictClassifierModel'
+    }
+    body.merge! opts: {
+      method: 'POST',
+      timeout: 15000
+    }
+    body.merge! params: {
+      ClassifierId: class_id,
+      Body: file,
+      BinaryToText: true
+    }
+
+    r = client.request(**body)
+    r.dig('Data', 'data').each_with_object({}) do |i, h|
+      h.merge! i['fieldName'] => i['fieldWordRaw']
+    end
+  end
+
   def ocr_223(file, class_id: 223)
     body = {
       action: 'PredictClassifierModel'