Update app.py

app.py

@@ -168,6 +168,7 @@ def main():
 
     if uploaded_file is not None:
         col1, col2 = st.columns(2)
+        col1.subheader("Sudoku Puzzle")
         puzzle = Image.open(uploaded_file)
         col1.image(puzzle, use_column_width=True)
         puzzle = np.asarray(puzzle)
@@ -203,13 +204,14 @@ def main():
         solve(grid)
     
         # Display the solution or appropriate message
-        if solve(grid):
-            with col2:
+         with col2:
+             if solve(grid):
+           
                 st.subheader("Sudoku Solved:")
-                display(grid)
+                st.table(grid)
             
-        else:
-            st.write("No solution found.")
+            else:
+                st.write("No solution found.")
 
 if __name__ == '__main__':
     main()