Spaces:
Running
Running
Pclanglais
commited on
Commit
•
cf17c33
1
Parent(s):
42051f9
Update app.py
Browse files
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 = "
|
19 |
ocr_llm = LLM(ocr_model_name, max_model_len=8128)
|
20 |
|
21 |
# Editorial Segmentation Model
|
22 |
-
editorial_model = "PleIAs/
|
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 |
)
|