Spaces:

Edgar404
/

Candy_Prototype

Sleeping

Edgar404 commited on Jun 7

Commit

a1df1d3

•

1 Parent(s): f93c354

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ base_processor = DonutProcessor.from_pretrained('Edgar404/donut-shivi-recognitio
 print('Loading complete')
 print('Loading the optimized model ....')
-optimized_model = VisionEncoderDecoderModel.from_pretrained('Edgar404/donut-shivi-cheques_KD_320', torch_dtype = torch.bfloat16 )
 optimized_processor = DonutProcessor.from_pretrained('Edgar404/donut-shivi-cheques_KD_320')
 print('Loading complete')

 print('Loading complete')
 print('Loading the optimized model ....')
+optimized_model = VisionEncoderDecoderModel.from_pretrained('Edgar404/donut-shivi-cheques_KD_320')
 optimized_processor = DonutProcessor.from_pretrained('Edgar404/donut-shivi-cheques_KD_320')
 print('Loading complete')