Hugging Face's logo

Spaces:
Edgar404
/
Candy_Prototype
Build error

App Files Community
Edgar404 commited on
Commit
a1df1d3
·
verified ·
1 Parent(s): f93c354

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ base_processor = DonutProcessor.from_pretrained('Edgar404/donut-shivi-recognitio
27
  print('Loading complete')
28
 
29
  print('Loading the optimized model ....')
30
- optimized_model = VisionEncoderDecoderModel.from_pretrained('Edgar404/donut-shivi-cheques_KD_320', torch_dtype = torch.bfloat16 )
31
  optimized_processor = DonutProcessor.from_pretrained('Edgar404/donut-shivi-cheques_KD_320')
32
  print('Loading complete')
33
 
 
27
  print('Loading complete')
28
 
29
  print('Loading the optimized model ....')
30
+ optimized_model = VisionEncoderDecoderModel.from_pretrained('Edgar404/donut-shivi-cheques_KD_320')
31
  optimized_processor = DonutProcessor.from_pretrained('Edgar404/donut-shivi-cheques_KD_320')
32
  print('Loading complete')
33