backend work for visualization stuff

app.py

@@ -73,6 +73,7 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         processed_frames = gr.Image(label="Live Frame")
+        # graphs = gr.Image(label="Graphs")
         original_frames = gr.Image(label="Original Frame")
 
     with gr.Row():

inference.py

@@ -100,20 +100,23 @@ palette = visualizer.dataset_meta.get('palette', None)
 
 print(len(classes))
 print(len(palette))
-def inference_frame_serial(image):
+def inference_frame_serial(image, visualize = True):
     start = time()
     result = inference_detector(model, image)
-    print(f'inference time: {time()-start}')
+    #print(f'inference time: {time()-start}')
     # show the results
-    visualizer.add_datasample(
-    'result',
-    image,
-    data_sample=result,
-    draw_gt = None,
-    show=False
-    )
-    frame = visualizer.get_image()
-    return frame
+    if visualize:
+        visualizer.add_datasample(
+        'result',
+        image,
+        data_sample=result,
+        draw_gt = None,
+        show=False
+        )
+        frame = visualizer.get_image()
+    else:
+        frame = None
+    return frame, result
 
 def inference_frame(image):
     result = inference_detector(model, image)
@@ -134,7 +137,6 @@ def inference_frame(image):
         cnt+=1
 
     #frames = process_frames(result, image, visualizer)
- 
     return frames
 
 def inference_frame_par_ready(image):