Ritvik19 commited on
Commit
413bacc
1 Parent(s): 039a0e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def echo_sudoku(sudoku, model_name):
32
  grid.flat[confidence_position] = confidence_value
33
  return puzzles[0]
34
 
35
- model_types = ['ffn', 'cnn', 'rnn', 'lstm', 'gru']
36
  model_sizes = ['64x2', '64x4', '128x2', '128x4']
37
  model_names = [f"{model_type}__{model_size}" for model_type in model_types for model_size in model_sizes]
38
 
 
32
  grid.flat[confidence_position] = confidence_value
33
  return puzzles[0]
34
 
35
+ model_types = ['ffn', 'cnn']
36
  model_sizes = ['64x2', '64x4', '128x2', '128x4']
37
  model_names = [f"{model_type}__{model_size}" for model_type in model_types for model_size in model_sizes]
38