Update app.py

app.py

@@ -6,19 +6,19 @@ import json
 def prediction(image):
 
     #we first use mp-02/layoutlmv3-finetuned-cord on the image, which gives us a JSON with some info and a blurred image
-    j, image_blurred = sroie_prediction(image)
+    j1, image_blurred = sroie_prediction(image)
     
     #then we use the model fine-tuned on sroie (for now it is Theivaprakasham/layoutlmv3-finetuned-sroie)
-    image = image_blurred.copy()
-    j2, image_final = cord_prediction(image)
+    img = image_blurred.copy()
+    j2, image_final = cord_prediction(img)
 
     #we then link the two json files
-    if len(d) == 0:
+    if len(j1) == 0:
         j3 = j2
     else:
-        j3 = json.dumps(j).split('}')[0] + ', ' + json.dumps(j2).split('{')[1]
+        j3 = json.dumps(j1).split('}')[0] + ', ' + json.dumps(j2).split('{')[1]
 
-    return j, image_blurred, j2, image_final, j3
+    return j1, image_blurred, j2, image_final, j3
 
 
 title = "Interactive demo: LayoutLMv3 for receipts"