Update app.py
Browse files
app.py
CHANGED
|
@@ -21,9 +21,9 @@ def ensure_list(x):
|
|
| 21 |
|
| 22 |
|
| 23 |
CHECKPOINTS = {
|
| 24 |
-
"LayoutLMv1
|
| 25 |
-
"LayoutLMv1 for Invoices
|
| 26 |
-
"Donut
|
| 27 |
}
|
| 28 |
|
| 29 |
PIPELINES = {}
|
|
@@ -226,13 +226,13 @@ CSS = """
|
|
| 226 |
}
|
| 227 |
.gradio-container .gr-button-primary {
|
| 228 |
background: linear-gradient(180deg, #FAED27 0%, #FAED27 100%);
|
| 229 |
-
border: 1px solid #
|
| 230 |
border-radius: 8px;
|
| 231 |
color: #1B8700;
|
| 232 |
}
|
| 233 |
.gradio-container.dark button#submit-button {
|
| 234 |
background: linear-gradient(180deg, #FAED27 0%, #FAED27 100%);
|
| 235 |
-
border: 1px solid #
|
| 236 |
border-radius: 8px;
|
| 237 |
color: #1B8700
|
| 238 |
}
|
|
|
|
| 21 |
|
| 22 |
|
| 23 |
CHECKPOINTS = {
|
| 24 |
+
"LayoutLMv1": "impira/layoutlm-document-qa",
|
| 25 |
+
"LayoutLMv1 for Invoices": "impira/layoutlm-invoices",
|
| 26 |
+
"Donut": "naver-clova-ix/donut-base-finetuned-docvqa",
|
| 27 |
}
|
| 28 |
|
| 29 |
PIPELINES = {}
|
|
|
|
| 226 |
}
|
| 227 |
.gradio-container .gr-button-primary {
|
| 228 |
background: linear-gradient(180deg, #FAED27 0%, #FAED27 100%);
|
| 229 |
+
border: 1px solid #000000;
|
| 230 |
border-radius: 8px;
|
| 231 |
color: #1B8700;
|
| 232 |
}
|
| 233 |
.gradio-container.dark button#submit-button {
|
| 234 |
background: linear-gradient(180deg, #FAED27 0%, #FAED27 100%);
|
| 235 |
+
border: 1px solid #000000;
|
| 236 |
border-radius: 8px;
|
| 237 |
color: #1B8700
|
| 238 |
}
|