Spaces:
Sleeping
Sleeping
A fix
Browse files
app.py
CHANGED
@@ -99,7 +99,10 @@ def inference(file_name, _batch_mode):
|
|
99 |
)
|
100 |
|
101 |
# exclude inference_start, inference_end
|
102 |
-
|
|
|
|
|
|
|
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))
|