Update app.py

app.py

@@ -17,13 +17,17 @@ en_model_path = "question_generator_by_en_on_pic"
 #zh_model_path = "question_generator_by_zh_on_pic"
 
 task_prompt = "<s_docvqa><s_question>{user_input}</s_question><s_answer>"
-en_pretrained_model = DonutModel.from_pretrained(en_model_path)
+#en_pretrained_model = DonutModel.from_pretrained(en_model_path)
 #zh_pretrained_model = DonutModel.from_pretrained(zh_model_path)
+en_pretrained_model = DonutModel.from_pretrained(en_model_path, ignore_mismatched_sizes=True)
 
 if torch.cuda.is_available():
     en_pretrained_model.half()
     device = torch.device("cuda")
     en_pretrained_model.to(device)
+else:
+    import torch
+    en_pretrained_model.encoder.to(torch.bfloat16)
 
 '''
 if torch.cuda.is_available():