Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,14 +26,14 @@ def classify(input_img):
|
|
| 26 |
id2label = {0: "invoice", 1: "receipt"}
|
| 27 |
return id2label[predicted_class_idx]
|
| 28 |
|
| 29 |
-
examples =[['https://upserve.com/media/sites/2/Bill-from-Mezcalero-in-Washington-D.C.-photo-by-Alfredo-Solis-1-e1507226752437.jpg',
|
| 30 |
-
|
| 31 |
demo = gr.Interface(
|
| 32 |
fn=classify,
|
| 33 |
inputs=gr.Image(shape=(200, 200)),
|
| 34 |
outputs="text",
|
| 35 |
allow_flagging="manual",
|
| 36 |
description="Upload an invoice or receipt image and the model will classify it!",
|
| 37 |
-
examples=examples
|
| 38 |
)
|
| 39 |
demo.launch(share=True)
|
|
|
|
| 26 |
id2label = {0: "invoice", 1: "receipt"}
|
| 27 |
return id2label[predicted_class_idx]
|
| 28 |
|
| 29 |
+
examples =[['https://upserve.com/media/sites/2/Bill-from-Mezcalero-in-Washington-D.C.-photo-by-Alfredo-Solis-1-e1507226752437.jpg'],
|
| 30 |
+
['https://templates.invoicehome.com/invoice-template-us-neat-750px.png'],]
|
| 31 |
demo = gr.Interface(
|
| 32 |
fn=classify,
|
| 33 |
inputs=gr.Image(shape=(200, 200)),
|
| 34 |
outputs="text",
|
| 35 |
allow_flagging="manual",
|
| 36 |
description="Upload an invoice or receipt image and the model will classify it!",
|
| 37 |
+
examples=examples,
|
| 38 |
)
|
| 39 |
demo.launch(share=True)
|