Update app.py

app.py

@@ -5,18 +5,19 @@ from tf_coder.value_search import value_search_settings
 import io
 from contextlib import redirect_stdout
 
-def get_problem():
-  """Specifies a problem to run TF-Coder on. Edit this function!"""
-  # A dict mapping input variable names to input tensors.
+
   inputs = {
       'rows': [10, 20, 30],
       'cols': [1, 2, 3, 4],
   }
-
+inputs = st.text_area('The input tensor(s) specified as key-value pairs', placeholder='''
+'rows': [10, 20, 30],
+'cols': [1,2,3,4],
+'''
   # The single desired output tensor.
   output = [[11, 12, 13, 14],
             [21, 22, 23, 24],
-            [31, 32, 33, 34]]
+            [31, 32, 33, 34]]  
 
   # A list of relevant scalar constants (if any).
   constants = []
@@ -37,7 +38,7 @@ settings = value_search_settings.from_dict({
 i = st.text_area("input tensor","second test")
 
 with io.StringIO() as buf, redirect_stdout(buf):
-    inputs, output, constants, description = get_problem()
-    colab_interface.run_value_search_from_colab(inputs, output, constants, description, settings)
+    _ , output, constants, description = get_problem()
+    results = colab_interface.run_value_search_from_colab(inputs, output, constants, description, settings)
     output = buf.getvalue()
-st.write(output)      
+st.code(output, language='bash')