Yehor commited on
Commit
1248b75
·
1 Parent(s): 441367e
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -99,7 +99,10 @@ def inference(file_name, _batch_mode):
99
  )
100
 
101
  # exclude inference_start, inference_end
102
- df = df.drop(["inference_start", "inference_end", "filename"])
 
 
 
103
 
104
  # round "inference_total" field to 2 decimal places
105
  df = df.with_columns(pl.col("inference_total").round(2))
 
99
  )
100
 
101
  # exclude inference_start, inference_end
102
+ if _batch_mode:
103
+ df = df.drop(["inference_start", "inference_end", "filenames"])
104
+ else:
105
+ df = df.drop(["inference_start", "inference_end", "filename"])
106
 
107
  # round "inference_total" field to 2 decimal places
108
  df = df.with_columns(pl.col("inference_total").round(2))