Update app.py

app.py

@@ -149,7 +149,7 @@ def run_model(input_text):
 
     #st.text_area(visualize (runtext,output))
     st.success(output)
-    return output
+   # return output
 
 
 col1, col2 = st.columns([1,1])
@@ -159,6 +159,19 @@ with col1:
     run_model(runtext)
     sentences=runtext.split('.')
     st.text_area('Reference text', str(reference_text))#,label_visibility="hidden")
+
+    with col1_tab1:
+        st.subheader("PROBLEM")
+                             
+
+    with col2_tab1:
+        st.subheader("DIAGNOSIS")
+                           
+    with col3_tab1:
+        st.subheader("MEDICATION")
+    with col4_tab1:
+        st.subheader("PROCEDURE")
+
 with col2:
     st.button('NER')
     doc = nlp(str(original_text2))
@@ -172,7 +185,7 @@ col3, col4 = st.columns(2)
 with col3: 
 #    st.text_area(visualize (runtext,output))
 
-    st.write(doc)
+    st.write('testing')
 with col4: 
     st.text_area('testing', str(reference_text))#,label_visibility="hidden")