Add application file

app.py

@@ -20,7 +20,7 @@ def run_donut(image):
     pixel_values = processor(image, return_tensors="pt").pixel_values.to(device)
     # pixel_values = processor(image, return_tensors="pt").pixel_values.to()
     task_prompt = '<passport_front>'
-    decoder_input_ids = processor.tokenizer(task_prompt, add_special_tokens=False, return_tensors="pt").input_ids.to("cuda")
+    decoder_input_ids = processor.tokenizer(task_prompt, add_special_tokens=False, return_tensors="pt").input_ids.to(device)
 
     outputs = model.generate(
         pixel_values=pixel_values,