Update app.py

app.py

@@ -38,12 +38,9 @@ if st.button("🔎 Search for Tensor Ops!"):
       'require_one_input_used': not settings_kwargs["require_all_inputs_used"],
   })
     with st.spinner("Searching for solution..."):
-        with io.StringIO() as buf, redirect_stdout(buf):
-            results = colab_interface.run_value_search_from_colab(i, o, c, description, settings)
-            stdout = buf.getvalue()
-        st.code(stdout, language='bash')
-        st.write(dir(results.benchmark))
+        results = colab_interface.run_value_search_from_colab(i, o, c, description, settings)
+        num_solutions = len(results.solutions)
+        st.write(f"Found {num_solutions} {"solutions" if num_solutions > 1 else "solution"} in {results.total_time} seconds")
         for solution in results.solutions:
             st.code(solution.expression, language='python')
-        st.write(results.total_time)
-        st.code(results.statistics)
+        st.write(dir(results))