starride-teklia commited on
Commit
92c11dc
1 Parent(s): 6b1f545

Update list of models

Browse files

Let the user choose between the models currently available in our PyLaia collection

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ IMAGE_ID_PATTERN = r"(?P<image_id>[-a-z0-9]{36})"
15
  CONFIDENCE_PATTERN = r"(?P<confidence>[0-9.]+)" # For line
16
  TEXT_PATTERN = r"\s*(?P<text>.*)\s*"
17
  LINE_PREDICTION = re.compile(rf"{IMAGE_ID_PATTERN} {CONFIDENCE_PATTERN} {TEXT_PATTERN}")
18
- models_name = ["Teklia/pylaia-rimes"]
 
19
  MODELS = {}
20
  DEFAULT_HEIGHT = 128
21
 
 
15
  CONFIDENCE_PATTERN = r"(?P<confidence>[0-9.]+)" # For line
16
  TEXT_PATTERN = r"\s*(?P<text>.*)\s*"
17
  LINE_PREDICTION = re.compile(rf"{IMAGE_ID_PATTERN} {CONFIDENCE_PATTERN} {TEXT_PATTERN}")
18
+ models_name = ["Teklia/pylaia-rimes", "Teklia/pylaia-belfort", "Teklia/pylaia-casia-hwdb2", "Teklia/pylaia-esposalles", "Teklia/pylaia-himanis", "Teklia/pylaia-home-alcar", "Teklia/pylaia-iam", "Teklia/pylaia-newseye-austrian", "Teklia/pylaia-norhand-v1", "Teklia/pylaia-norhand-v2", "Teklia/pylaia-norhand-v3", "Teklia/pylaia-popp", "Teklia/pylaia-rimes", "Teklia/pylaia-PELLET-CasimirMarius"]
19
+ ]
20
  MODELS = {}
21
  DEFAULT_HEIGHT = 128
22