use KBLab's swedish-ocr-correction model (#1)
Browse files- use KBLab's swedish-ocr-correction model (4583cb1f2e72924f5df8e2a424a7f47660e494df)
Co-authored-by: Robin Qrtz <[email protected]>
app.py
CHANGED
@@ -22,7 +22,7 @@ examples = {
|
|
22 |
# Load model
|
23 |
@st.cache_resource
|
24 |
def load_model():
|
25 |
-
return T5ForConditionalGeneration.from_pretrained('
|
26 |
model = load_model()
|
27 |
|
28 |
|
|
|
22 |
# Load model
|
23 |
@st.cache_resource
|
24 |
def load_model():
|
25 |
+
return T5ForConditionalGeneration.from_pretrained('KBLab/swedish-ocr-correction')
|
26 |
model = load_model()
|
27 |
|
28 |
|