Update app.py

app.py

@@ -1,14 +1,7 @@
-from inference import prediction as cord_sroie_prediction
+from inference import prediction
 import gradio as gr
 import json
 
-def prediction(image):
-
-    j2, image_final = cord_sroie_prediction(image)
-
-    return j2, image_final
-
-
 title = "Interactive demo: LayoutLMv3 for receipts"
 description = "Demo for Microsoft's LayoutLMv3, a Transformer for state-of-the-art document image understanding tasks. This particular space uses two instances of the model, one fine-tuned on CORD and the other SROIE.\n It firsts uses the fine-tune on SROIE to extract date, company and address, then the fine-tune on CORD for the other info. To use it, simply upload an image or use the example image below. Results will show up in a few seconds."
 examples = [['image.jpeg'],['image.png']]