vat

app.py

@@ -39,9 +39,9 @@ def read_root(image_file: bytes = File(...)):
     image = Image.open(BytesIO(image_file))
 
     question_1 = "What is the Total amount?"
-    question_2 = "What is Total VAT amount?"
+    question_2 = "What is the VAT amount?"
     question_3 = "What is the Date?"
-    question_4 = "What is the total amount currency?"
+    question_4 = "What is the tax amount"
     output_1 = pipe(image, question_1)
     output_2 = pipe(image, question_2)
     output_3 = pipe(image, question_3)
@@ -49,7 +49,7 @@ def read_root(image_file: bytes = File(...)):
     
     response = {}
     response['total amount'] = output_1[0]['answer']
-    response['toal vat'] = output_2[0]['answer']
+    response['total vat'] = output_2[0]['answer']
     response['date'] = output_3[0]['answer']
-    response['currency'] = output_4[0]['answer']
+    response['vat'] = output_4[0]['answer']
     return response