Update app.py

app.py

@@ -31,7 +31,7 @@ Follow them on social networks and **contact** if you need any help or have any
 """.strip()
 
 examples = [
-    ["evaluation_results.jsonl", True],
+    ["evaluation_results.jsonl", True, False],
 ]
 
 description_head = f"""
@@ -65,10 +65,12 @@ tech_libraries = f"""
 - polars: {version('polars')}
 """.strip()
 
+
 def clean_value(x):
     return x.replace('’', "'").strip().lower().replace(',', '').replace('.', '').replace('?', '').replace('!', '').replace('–', '').replace('«', '').replace('»', '')
 
-def inference(file_name, clear_punctuation, progress=gr.Progress()):
+    
+def inference(file_name, clear_punctuation, show_chars, progress=gr.Progress()):
     if not file_name:
         raise gr.Error("Please paste your JSON file.")
 
@@ -116,6 +118,16 @@ def inference(file_name, clear_punctuation, progress=gr.Progress()):
     results.append('')
     results.append(f"- RTF: {round(rtf, 4)}")
 
+    if show_chars:
+        all_chars = set()
+        for pred in list(df['prediction']):
+            for c in pred:
+                all_chars.add(c)
+
+        results.append('')
+        results.append(f"Chars in predictions:")
+        results.append(f"{list(all_chars)}")
+
     return "\n".join(results)
 
 
@@ -136,6 +148,9 @@ with demo:
             clear_punctuation = gr.Checkbox(
                 label="Clear punctuation, some chars and convert to lowercase",
             )
+            show_chars = gr.Checkbox(
+                label="Show chars in predictions",
+            )
 
         metrics = gr.Textbox(
             label="Metrics",
@@ -151,7 +166,7 @@ with demo:
     )
 
     with gr.Row():
-        gr.Examples(label="Choose an example", inputs=[jsonl_file, clear_punctuation], examples=examples)
+        gr.Examples(label="Choose an example", inputs=[jsonl_file, clear_punctuation, show_chars], examples=examples)
 
     gr.Markdown(description_foot)