sudacode/owocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Silence warning in accurate filtering

Browse Source
This commit is contained in:
AuroraWright
2025-04-22 18:45:27 -07:00
parent 41f9c60dfb
commit 010f319672
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
owocr/run.py
View File

@@ -6,6 +6,7 @@ from pathlib import Path
import queue import queue
import io import io
import re import re
import logging
import inspect import inspect
import numpy as np import numpy as np
@@ -299,6 +300,7 @@ class TextFiltering:
try: try:
from transformers import pipeline, AutoTokenizer from transformers import pipeline, AutoTokenizer
import torch import torch
logging.getLogger('transformers').setLevel(logging.ERROR)
model_ckpt = 'papluca/xlm-roberta-base-language-detection' model_ckpt = 'papluca/xlm-roberta-base-language-detection'
tokenizer = AutoTokenizer.from_pretrained( tokenizer = AutoTokenizer.from_pretrained(