Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -20,5 +20,5 @@ task_prompt = f"<s_{task_name}>"
|
|
20 |
|
21 |
model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
|
22 |
model.eval()
|
23 |
-
demo = gr.Interface(fn=demo_process,inputs=gr.
|
24 |
demo.launch()
|
|
|
20 |
|
21 |
model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
|
22 |
model.eval()
|
23 |
+
demo = gr.Interface(fn=demo_process,inputs=gr.File(label="Upload PDF"),outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
|
24 |
demo.launch()
|