Update app.py

app.py

@@ -1,9 +1,11 @@
 from PIL import Image, ImageDraw, ImageFont
 from dotenv import load_dotenv
 import matplotlib.pyplot as plt
+from io import BytesIO
 import gradio as gr
 import numpy as np
 import requests
+import base64
 import boto3
 import uuid
 import os
@@ -154,7 +156,9 @@ def testing_yolocounter(input_img):
     countings = response['countings_scinames']
     coverage = response['coverage']
     detections = response['detections']
-    img, text = display_detectionsandcountings_yolocounter(input_img, detections, countings, coverage, prob_th=0, cth = 0)
+    img_out = response['img_out']
+    img = Image.open(BytesIO(base64.b64decode(img_out)))
+    _, text = display_detectionsandcountings_yolocounter(input_img, detections, countings, coverage, prob_th=0, cth = 0)
     return img, text
 
 def testing_directcounter(input_img):