Update app.py

app.py

@@ -12,14 +12,12 @@ from donut import DonutModel
 
 def demo_process(input_img):
     global pretrained_model, task_prompt, task_name
-    # input_img = Image.fromarray(input_img)
     output = pretrained_model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
     return output
 
 task_prompt = f"<s_data-donut_mixed>"
 
 pretrained_model = DonutModel.from_pretrained("RustX/donut-base-finetuned-insurance")
-#pretrained_model.encoder.to(torch.bfloat16)
 pretrained_model.eval()
 
 demo = gr.Interface(
@@ -28,7 +26,7 @@ demo = gr.Interface(
     outputs="json",
     title=f"Donut 🍩 demonstration for `insurance` task",
     description="""This model is trained with 294 Korean insurance images of custom dataset.""",
-    examples=[["donut_sample_insurance1.jpg"], ["donut_sample_insurance2.jpg"]],
+    #examples=[["donut_sample_insurance1.jpg"], ["donut_sample_insurance2.jpg"]],
     cache_examples=False,
 )