Update app.py

modiy '_add_markup': use the raw table if parsing fails

app.py

@@ -10,15 +10,19 @@ from peft import PeftModel
 ## CoT prompts
 
 def _add_markup(table):
-    parts = [p.strip() for p in table.splitlines(keepends=False)]
-    if parts[0].startswith('TITLE'):
-        result = f"Title: {parts[0].split(' | ')[1].strip()}\n"
-        rows = parts[1:]
-    else:
-        result = ''
-        rows = parts
-    prefixes = ['Header: '] + [f'Row {i+1}: ' for i in range(len(rows) - 1)]
-    return result + '\n'.join(prefix + row for prefix, row in zip(prefixes, rows))
+    try:
+        parts = [p.strip() for p in table.splitlines(keepends=False)]
+        if parts[0].startswith('TITLE'):
+            result = f"Title: {parts[0].split(' | ')[1].strip()}\n"
+            rows = parts[1:]
+        else:
+            result = ''
+            rows = parts
+        prefixes = ['Header: '] + [f'Row {i+1}: ' for i in range(len(rows) - 1)]
+        return result + '\n'.join(prefix + row for prefix, row in zip(prefixes, rows))
+    except:
+        # just use the raw table if parsing fails
+        return table
 
 
 _TABLE = """Year | Democrats | Republicans | Independents
@@ -267,5 +271,4 @@ with gr.Blocks(theme=theme) as demo:
         process_document, inputs=[input_image, instruction, llm], outputs=[output_table, output_text]
     )
 
-demo.queue()
-demo.launch()
+demo.queue(concurrency_count=1).launch()