Hugging Face's logo

Spaces:
RustX
/
donut-base-finetuned-insurance
Runtime error

App Files Community
RustX commited on
Commit
10a4c4e
·
1 Parent(s): dbf659a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def demo_process(input_img):
19
  task_prompt = f"<s_data-donut_mixed>"
20
 
21
  pretrained_model = DonutModel.from_pretrained("RustX/donut-base-finetuned-insurance")
22
- pretrained_model.encoder.to(torch.bfloat16)
23
  pretrained_model.eval()
24
 
25
  demo = gr.Interface(
 
19
  task_prompt = f"<s_data-donut_mixed>"
20
 
21
  pretrained_model = DonutModel.from_pretrained("RustX/donut-base-finetuned-insurance")
22
+ #pretrained_model.encoder.to(torch.bfloat16)
23
  pretrained_model.eval()
24
 
25
  demo = gr.Interface(