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 |
}
|