improve gradio blocks interface

app.py

@@ -12,8 +12,12 @@ import matplotlib.patches as patches
 from matplotlib.patches import Polygon  
 import numpy as np
 import random
+import json
 
 
+with open("config.json", "r") as f:
+    config = json.load(f)
+
 d_model = config['text_config']['d_model']
 num_layers = config['text_config']['encoder_layers']
 attention_heads = config['text_config']['encoder_attention_heads']