use KBLab's swedish-ocr-correction model

#1
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