Commit e7506d38 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): fix csv read error

parent 5f0835a1

imgutils/generic/multilabel_timm.py

+2 −2
Original line number Diff line number Diff line
@@ -142,7 +142,7 @@ class MultiLabelTIMMModel: 
                    repo_type='model',

                    filename='selected_tags.csv',

                    token=self._get_hf_token(),

                ))

                ), keep_default_na=False)



                with open(hf_hub_download(

                        repo_id=self.repo_id,
@@ -193,7 +193,7 @@ class MultiLabelTIMMModel: 
                        repo_id=self.repo_id,

                        repo_type='model',

                        filename='thresholds.csv'

                    ))

                    ), keep_default_na=False)

                except (EntryNotFoundError,):

                    self._default_category_thresholds = {}

                else: