Spaces:
Running
Running
Use all metrics
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ tech_libraries = f"""
|
|
64 |
- evaluate: {version("evaluate")}
|
65 |
- pandas: {version("pandas")}
|
66 |
- polars: {version("polars")}
|
67 |
-
-
|
68 |
""".strip()
|
69 |
|
70 |
|
@@ -160,23 +160,28 @@ def process_file(file_name, _batch_mode, _calculate_distance, _calculate_metrics
|
|
160 |
]
|
161 |
|
162 |
df = pl.DataFrame(df_pd)
|
163 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
164 |
df = df.with_columns(
|
165 |
pld.col("prediction").dist_str.levenshtein("reference").alias("distance")
|
166 |
)
|
167 |
|
|
|
168 |
fields = [
|
169 |
"distance",
|
170 |
-
|
171 |
-
"reference",
|
172 |
-
]
|
173 |
-
else:
|
174 |
-
fields = [
|
175 |
-
"prediction",
|
176 |
-
"reference",
|
177 |
]
|
178 |
|
179 |
-
|
|
|
|
|
180 |
|
181 |
|
182 |
demo = gr.Blocks(
|
|
|
64 |
- evaluate: {version("evaluate")}
|
65 |
- pandas: {version("pandas")}
|
66 |
- polars: {version("polars")}
|
67 |
+
- polars-distance: {version("polars_distance")}
|
68 |
""".strip()
|
69 |
|
70 |
|
|
|
160 |
]
|
161 |
|
162 |
df = pl.DataFrame(df_pd)
|
163 |
+
else:
|
164 |
+
fields = [
|
165 |
+
"prediction",
|
166 |
+
"reference",
|
167 |
+
]
|
168 |
+
|
169 |
+
df = df.select(fields)
|
170 |
+
|
171 |
+
if _calculate_distance:
|
172 |
df = df.with_columns(
|
173 |
pld.col("prediction").dist_str.levenshtein("reference").alias("distance")
|
174 |
)
|
175 |
|
176 |
+
# add distance to the first position
|
177 |
fields = [
|
178 |
"distance",
|
179 |
+
*fields,
|
|
|
|
|
|
|
|
|
|
|
|
|
180 |
]
|
181 |
|
182 |
+
df = df.select(fields)
|
183 |
+
|
184 |
+
return df
|
185 |
|
186 |
|
187 |
demo = gr.Blocks(
|