Spaces:
Running
Running
Restore working Language enum implementation
Browse files- 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 |
-
|
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 |
-
|
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
|