Update app.py

app.py

@@ -6,9 +6,9 @@ from contextlib import redirect_stdout
 from streamlit_ace import st_ace
 st.set_page_config(page_icon='👩‍💻', layout="wide")
 
-st.write('## Inputs')
+st.write('### Inputs')
 inputs = st_ace(placeholder="The input tensor(s) specified as a dictionary", value="{'rows': [10, 20, 30],\n'cols': [1,2,3,4]}", language="python", theme="solarized_dark")
-st.write('## Outputs')
+st.write('### Outputs')
 output = st_ace(placeholder="The output tensor", value="[[11, 12, 13, 14],\n[21, 22, 23, 24],\n[31, 32, 33, 34]]", language="python", theme="solarized_dark")
 
 st.sidebar.header("Settings:")
@@ -31,6 +31,6 @@ with io.StringIO() as buf, redirect_stdout(buf):
 
     description = 'add two vectors with broadcasting to get a matrix'
     inputs = {'rows': [10, 20, 30],'cols': [1,2,3,4]}
-    results = colab_interface.run_value_search_from_colab(eval(inputs), eval(output), constants, description, settings)
+    results = colab_interface.run_value_search_from_colab(eval(inputs), eval(output.replace('\n',''), constants, description, settings)
     stdout = buf.getvalue()
 st.code(stdout, language='bash')