Update app.py

app.py

@@ -2,7 +2,7 @@ import time
 import streamlit as st
 import torch
 import string
-import json
+
 
 from transformers import BertTokenizer, BertForMaskedLM
 
@@ -91,8 +91,8 @@ try:
           start = time.time()
           try:
             res = get_bert_prediction(input_text,top_k)
-            st.header("JSON:")
-            st.json(json.dumps(res,indent=4))
+            st.caption("Results in JSON")
+            st.json(res)
             
           except Exception as e:
             st.error("Some error occurred during prediction" + str(e))