Spaces:

unstructuredio
/

receipt-parser

Runtime error

laverdes commited on Jan 6, 2023

Commit

c3a6b55

1 Parent(s): ef55841

fix: comment encoder bfloat16

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,8 +73,8 @@ pretrained_model = VisionEncoderDecoderModel.from_pretrained("unstructuredio/don
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pretrained_model.to(device)
-pretrained_model.encoder.to(torch.bfloat16)
-pretrained_model.eval()
 st.text(f'parsing receipt..')
 parsed_receipt_info = run_prediction(image)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 pretrained_model.to(device)
+# pretrained_model.encoder.to(torch.bfloat16)
+# pretrained_model.eval()
 st.text(f'parsing receipt..')
 parsed_receipt_info = run_prediction(image)