Update app.py

app.py

@@ -10,17 +10,6 @@ inputs = st.text_area('The input tensor(s) specified as key-value pairs', placeh
 'cols': [1,2,3,4],
 '''
   # The single desired output tensor.
-  output = [[11, 12, 13, 14],
-            [21, 22, 23, 24],
-            [31, 32, 33, 34]]  
-
-  # A list of relevant scalar constants (if any).
-  constants = []
-
-  # An English description of the tensor manipulation.
-  description = 'add two vectors with broadcasting to get a matrix'
-
-  return inputs, output, constants, description
 
 settings = value_search_settings.from_dict({
       'timeout': 300,
@@ -33,7 +22,16 @@ settings = value_search_settings.from_dict({
 i = st.text_area("input tensor","second test")
 
 with io.StringIO() as buf, redirect_stdout(buf):
+    output = [[11, 12, 13, 14],
+            [21, 22, 23, 24],
+            [31, 32, 33, 34]]  
+
+  # A list of relevant scalar constants (if any).
+    constants = []
+
+  # An English description of the tensor manipulation.
+    description = 'add two vectors with broadcasting to get a matrix'
     _ , output, constants, description = get_problem()
     results = colab_interface.run_value_search_from_colab(inputs, output, constants, description, settings)
-    output = buf.getvalue()
-st.code(output, language='bash')   
+    stdout = buf.getvalue()
+st.code(stdout, language='bash')