viklofg robinq commited on
Commit
7c2d88d
1 Parent(s): aa4704a

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]>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ examples = {
22
  # Load model
23
  @st.cache_resource
24
  def load_model():
25
- return T5ForConditionalGeneration.from_pretrained('viklofg/swedish-ocr-correction')
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