Update app.py

app.py

@@ -14,7 +14,16 @@ parser.add_argument("--pretrained_path", type=str, default="uartimcs/donut-booki
 args, left_argv = parser.parse_known_args()
 task_name = args.task
 task_prompt = f"<s_{task_name}>"
+
+image = Image.open("./sample-booking/CMA_150.jpg")
+image.save("CMA_sample.jpg")
+image = Image.open("./sample-booking/COSCO_150.jpg")
+image.save("COSCO_sample.jpg")
+image = Image.open("./sample-booking/ONEY_150.jpg")
+image.save("ONEY_sample.jpg")
+
+
 model = DonutModel.from_pretrained("uartimcs/donut-booking-extract")
 model.eval()
-demo = gr.Interface(fn=demo_process,inputs="image",outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
+demo = gr.Interface(fn=demo_process,inputs=gr.inputs.Image(type="pil"),outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task", examples=[["CMA_sample.jpg"], ["COSCO_sample.jpg"], ["ONEY_sample.jpg"]],)
 demo.launch()