error runing test.py file in cpu mode.

[ajberenstein]

Hi, first of all let me congratulate you for this fantastic work.
After following the installation instructions I tried to run the usage example in cpu_mode
python test.py --imgfile=example/sample.jpg --use_cpu=1
and I get the following error:
[E 230209 19:16:52 test:125] Unexpected error has occurred
Traceback (most recent call last):
File "test.py", line 123, in
main(args)
File "test.py", line 88, in main
test(
File "test.py", line 53, in test
vd, rec_img = vectorize(pil_img, inps, outs)
File "/home/ariel/repos/derendering-text/src/modules/postprocess/vector.py", line 114, in vectorize
tb_param, effect_param, effect_visibility = extract_rendering_params(
File "/home/ariel/repos/derendering-text/src/modules/postprocess/manipulate.py", line 286, in extract_rendering_params
tb_param = get_textblob_param_with_affine(
File "/home/ariel/repos/derendering-text/src/modules/postprocess/rpe.py", line 515, in get_textblob_param_with_affine
text_form_dto = TextFormData(
TypeError: init() missing 1 required positional argument: 'text_index'

what I can see in the file dto_skia.py, is that the dataclass TextFromData requires a text_index: int value that is not set in rpe.py file. Please, let me know if you need more info in order to reproduce this error.
Best,
A.

[ajberenstein]

Udate: after cloning the whole repo, installing requirements, upload provided weights and generating util_lib.gen_pams, the tes.py code (in GPU mode) ends with the same error.

Session Info:
['Python 3.8.10',
'torch==1.13.1+cu116',
'torchviosion==0.14.1+cu116',
'h5py==3.1.0',
'skia-python==87.5',
'fonttools==4.38.0',
'logzero==1.7.0',
'opencv-python==4.6.0.66',
'pyclipper==1.3.0.post4',
'scikit-image==0.18.3',
'scikit-learn==1.0.2',
'scipy==1.7.3']

[shimoda-uec]

Thank you for reporting the error.
I fixed it.

[ajberenstein]

Thanks Shimoda, It worked.
best,
A.