Update app.py

app.py

@@ -34,6 +34,8 @@ def process_document(image,sendimg):
         im1 = Image.fromarray(image)
     elif sendimg == False:
         im1 = Image.open('./no_image.jpg')
+    #keep track of demo count
+    resp = requests.get('https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fto-be%2Finvoice_document_headers_extraction_with_donut%2Fdemo&label=demos%20served&labelColor=%23edd239&countColor=%23d9e3f0')
     
     #send notification through telegram
     TOKEN = os.getenv('TELEGRAM_BOT_TOKEN')
@@ -85,6 +87,8 @@ paragraph3 = '<p><strong>Benchmark observations:</strong><br />From all document
 paragraph4 = '<p><strong>Try it out:</strong><br />To use it, simply upload your image and click &#39;submit&#39;, or click one of the examples to load them.<br /><em>(because this is running on the free cpu tier, it will take about 40 secs before you see a result. On a GPU it takes less than 2 seconds)</em></p><p>&nbsp;</p><p>Have fun&nbsp;😎</p><p>Toon Beerten</p>'
 smallprint = '<p>✤&nbsp;<span style="font-size:11px">To get an idea of the usage, you can opt to let me get personally notified via Telegram with the image uploaded. All data will be automatically deleted after 48 hours</span></p>'
 css = "#inp {height: auto !important; width: 100% !important;}"
+visit_badge = '<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fto-be%2Finvoice_document_headers_extraction_with_donut"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fto-be%2Finvoice_document_headers_extraction_with_donut&labelColor=%23edd239&countColor=%23d9e3f0&style=flat" /></a>'
+
 # css = "@media screen and (max-width: 600px) { .output_image, .input_image {height:20rem !important; width: 100% !important;} }"
 # css = ".output_image, .input_image {height: 600px !important}"
 
@@ -123,6 +127,8 @@ with gr.Blocks(css=css) as demo:
     btn.click(fn=process_document, inputs=[inp,sendimg], outputs=[jsonout,imgout])
 
     gr.HTML(smallprint)
+    gr.HTML(visit_badge)
+    
 
 demo.launch()