Update app.py

app.py

@@ -40,6 +40,7 @@ def reframe(points):
     points_new[2] = points[np.argmax(diff)]
     return points_new
 
+
 def splitcells(img):
     rows = np.vsplit(img,9)
     boxes = []
@@ -49,6 +50,7 @@ def splitcells(img):
             boxes.append(box)
     return boxes
 
+
 def CropCell(cells):
     Cells_croped = []
     for image in cells:
@@ -60,6 +62,7 @@ def CropCell(cells):
         
     return Cells_croped
 
+
 def read_cells(cell,model):
 
     result = []
@@ -73,7 +76,7 @@ def read_cells(cell,model):
         # getting predictions and setting the values if probabilities are above 65% 
         
         predictions = model.predict(img)
-        classIndex = np.argmax(model.predict(img), axis=-1)
+        classIndex = np.argmax(predictions, axis=1)
         probabilityValue = np.amax(predictions)
         
         if probabilityValue > 0.65:
@@ -82,6 +85,8 @@ def read_cells(cell,model):
             result.append(0)
     return result
 
+
+
 #This function finds the next box to solve 
 
 def next_box(quiz):
@@ -141,9 +146,6 @@ def Solved(quiz):
             else:
                 print(str(quiz[row][col]) + " ", end="")
                 
-
-            
-
 def main():
     st.title('Sudoku Solver')
 
@@ -154,15 +156,14 @@ def main():
         puzzle = Image.open(uploaded_file)
         col1.image(puzzle, use_column_width=True)
         puzzle = np.asarray(puzzle)
-     
         # Resizing puzzle to be solved
-        sudoku_a = cv2.resize(puzzle, (450,450))
+        puzzle = cv2.resize(puzzle, (450,450))
         # Preprocessing Puzzle 
-        su_puzzle = preprocess(sudoku_a)
+        su_puzzle = preprocess(puzzle)
         
         # Finding the outline of the sudoku puzzle in the image
         su_contour_1= su_puzzle.copy()
-        su_contour_2= sudoku_a.copy()
+        su_contour_2= puzzle.copy()
         su_contour, hierarchy = cv2.findContours(su_puzzle,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
         cv2.drawContours(su_contour_1, su_contour,-1,(0,255,0),3)
         
@@ -183,7 +184,7 @@ def main():
         grid = read_cells(sudoku_cell_croped, model)
         grid = np.asarray(grid)
         grid = np.reshape(grid,(9,9))
-        solve(grid)    
+        solve(grid)
     
         # Display the solution or appropriate message
         if solve(grid):