Update OCR.py

OCR.py

@@ -14,7 +14,7 @@ processor = AutoProcessor.from_pretrained(model_id)
 
 def extract_text_from_image(image_content):
     image = Image.open(io.BytesIO(image_content))
-    prompt = "Extract all relevant details from this invoice."
+    prompt =  "Extract the following details from this invoice: Invoice Number, Amount, Invoice Date, Due Date."
     inputs = processor(text=prompt, images=image, return_tensors="pt").to(device)
     input_len = inputs["input_ids"].shape[-1]
     with torch.inference_mode():