Pclanglais commited on
Commit
cf17c33
1 Parent(s): 42051f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,11 +15,11 @@ import difflib
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  # OCR Correction Model
18
- ocr_model_name = "Pclanglais/ocronos2"
19
  ocr_llm = LLM(ocr_model_name, max_model_len=8128)
20
 
21
  # Editorial Segmentation Model
22
- editorial_model = "PleIAs/Estienne"
23
  token_classifier = pipeline(
24
  "token-classification", model=editorial_model, aggregation_strategy="simple", device=device
25
  )
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  # OCR Correction Model
18
+ ocr_model_name = "PleIAs/OCRonos"
19
  ocr_llm = LLM(ocr_model_name, max_model_len=8128)
20
 
21
  # Editorial Segmentation Model
22
+ editorial_model = "PleIAs/Segmentext"
23
  token_classifier = pipeline(
24
  "token-classification", model=editorial_model, aggregation_strategy="simple", device=device
25
  )