milwright commited on
Commit
6931bb0
·
1 Parent(s): 4edf687

Restore working Language enum implementation

Browse files
Files changed (1) hide show
  1. structured_ocr.py +2 -10
structured_ocr.py CHANGED
@@ -137,19 +137,11 @@ class LanguageMeta(Enum.__class__):
137
  def __new__(metacls, cls, bases, classdict):
138
  languages = get_language_dict()
139
  for code, name in languages.items():
140
- key = name.upper().replace(' ', '_')
141
- # Skip if key already exists in the fallback languages
142
- if key not in classdict:
143
- classdict[key] = name
144
  return super().__new__(metacls, cls, bases, classdict)
145
 
146
  class Language(Enum, metaclass=LanguageMeta):
147
- # Fallback languages in case enum creation fails
148
- ENGLISH = "English"
149
- SPANISH = "Spanish"
150
- FRENCH = "French"
151
- GERMAN = "German"
152
- LATIN = "Latin"
153
 
154
  class StructuredOCRModel(BaseModel):
155
  file_name: str
 
137
  def __new__(metacls, cls, bases, classdict):
138
  languages = get_language_dict()
139
  for code, name in languages.items():
140
+ classdict[name.upper().replace(' ', '_')] = name
 
 
 
141
  return super().__new__(metacls, cls, bases, classdict)
142
 
143
  class Language(Enum, metaclass=LanguageMeta):
144
+ pass
 
 
 
 
 
145
 
146
  class StructuredOCRModel(BaseModel):
147
  file_name: str