update layout

app.py

@@ -12,9 +12,9 @@ import gradio as gr
 from huggingface_hub import snapshot_download
 from prettytable import PrettyTable
 import pandas as pd
+import torch
 
 config = {
-    "input": "demo",
     "model_type": "roberta",
     "model_name_or_path": "roberta-large",
     "logic_lambda": 0.5,
@@ -24,7 +24,9 @@ config = {
     "max_seq1_length": 256,
     "max_seq2_length": 128,
     "max_num_questions": 8,
-    "do_lower_case": False
+    "do_lower_case": False,
+    "seed": 42,
+    "n_gpu": torch.cuda.device_count(),
 }
 
 os.system('git clone https://github.com/jiangjiechen/LOREN/')
@@ -80,7 +82,9 @@ def gradio_formatter(js, output_type):
         pt.add_row(v)
 
     html = pt.get_html_string(attributes={
-        'style': 'border: 1; border-width: 1px; border-collapse: collapse; align: left',
+        'style': 'border-width: 1px; border-collapse: collapse',
+        'align': 'left',
+        'border': '1'
     }, format=True)
     return html