roboflow / inference

A fast, easy-to-use, production-ready inference server for computer vision supporting deployment of many popular model architectures and fine-tuned models.

https://inference.roboflow.com

Other

1.13k stars 84 forks source link

TrOCR in Inference #438

Open stellasphere opened 1 month ago

stellasphere commented 1 month ago

Description

Adds TrOCR, a transformer based OCR model, to Inference.
Alters the DoctrOCRInferenceResponse to OCRInferenceResponse to have a consistent output format for all OCR related tasks

Type of change

[x] New feature (non-breaking change which adds functionality)

How has this change been tested, please provide a testcase or example of how you tested the change?

Locally, on M1

Any specific deployment considerations

Docs

[ ] Docs updated? What were the changes:

probicheaux commented 1 month ago

To fix code quality:

pip install -r requirements/requirements.code_analysis.txt
make style

probicheaux commented 1 week ago

@stellasphere can we try to get this in?

PawelPeczek-Roboflow commented 1 week ago

Error in server start - this is probably why tests are failing

File "/app/cpu_http.py", line 2, in <module>
    from inference.core.interfaces.http.http_api import HttpInterface
  File "/app/inference/core/interfaces/http/http_api.py", line 67, in <module>
    from inference.core.entities.responses.paligemma import PaliGemmaInferenceResponse
ModuleNotFoundError: No module named 'inference.core.entities.responses.paligemma'

PawelPeczek-Roboflow commented 1 week ago

I believe one of the main merge failed to resolve all conflicts