Added plt.close() to help with resource management

app.py

@@ -301,6 +301,7 @@ def count(image, text, prompts, state, device):
     plt.axis('off')
     img_buf = io.BytesIO()
     plt.savefig(img_buf, format='png', bbox_inches='tight')
+    plt.close()
 
     output_img = Image.open(img_buf)
 
@@ -386,6 +387,7 @@ def count_main(image, text, prompts, device):
     plt.axis('off')
     img_buf = io.BytesIO()
     plt.savefig(img_buf, format='png', bbox_inches='tight')
+    plt.close()
 
     output_img = Image.open(img_buf)