Decode image from base64

app.py

@@ -54,11 +54,15 @@ def predict_image(img, threshold):
         )
             
     json_data = json.loads(response.content)
-    pil_img = Image.fromarray(json_data['image'])
+    image_data = json_data['image']
     shape = json_data['shape']
     infos = json_data['infos']
     duration = json_data['duration']
 
+    img_data_in = base64.b64decode(json_data['image']) 
+    np_img = np.frombuffer(img_data_in, dtype=np.uint8).reshape(shape)
+    pil_img = Image.fromarray(np_img)
+
     return pil_img, shape, infos, duration
 
 
@@ -87,7 +91,7 @@ TITLE = "Oriented bounding boxes detection on Optical Satellite images"
 # Define the Gradio Interface
 demo = gr.Blocks(title=TITLE, css=css).queue()
 with demo:
-    gr.Markdown(f"<h3><center>{TITLE}<center><h3>")
+    gr.Markdown(f"<h1><center>{TITLE}<center><h1>")
 
     with gr.Row():
         with gr.Column(scale=0):