Update app.py

app.py

@@ -314,6 +314,7 @@ class DetectionPredictor(BasePredictor):
                         #header = ['id', 'class', 'speed', 'weight']
                         #writer2.writerow(header)
         img= draw_boxes(im0, bbox_xyxy, self.model.names, object_id,writer, writer2, identities)
+        cv2.imshow("window", img)
         #df = pd.read_csv("/pulse.csv")
         #df['time'] = pd.to_datetime(df['time'], format = '%H:%M:%S  %d.%m.%Y')
         
@@ -334,7 +335,7 @@ class DetectionPredictor(BasePredictor):
         #except:
           #log_string += f'An error occured while saving figure_{num:010d}.png, '
       
-        return log_string, img
+        return log_string
 
 
 @hydra.main(version_base=None, config_path=str(DEFAULT_CONFIG.parent), config_name=DEFAULT_CONFIG.name)