Faustrix commited on
Commit
feafc9d
1 Parent(s): d425ac5

Update app.py with improved Gradio interface and update model description

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoModel, AutoTokenizer
5
  model = AutoModel.from_pretrained("openbmb/MiniCPM-V-2", trust_remote_code=True)
6
 
7
  # Load the tokenizer
8
- tokenizer = AutoTokenizer.from_pretrained("openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True)
9
 
10
  model.eval()
11
 
@@ -15,7 +15,7 @@ question = gr.Textbox(value="Using the standard 9x9 sudoku format, solve the sud
15
  answer = gr.Textbox(label="Answer", show_label=True, show_copy_button=True)
16
 
17
  title = "Sudoku Solver by FG"
18
- description = "Sudoku Solver using MiniCPM-Llama3-V-2_5"
19
 
20
  # Define the function for solving Sudoku
21
  def solve_sudoku(image, question):
 
5
  model = AutoModel.from_pretrained("openbmb/MiniCPM-V-2", trust_remote_code=True)
6
 
7
  # Load the tokenizer
8
+ tokenizer = AutoTokenizer.from_pretrained("openbmb/MiniCPM-V-2", trust_remote_code=True)
9
 
10
  model.eval()
11
 
 
15
  answer = gr.Textbox(label="Answer", show_label=True, show_copy_button=True)
16
 
17
  title = "Sudoku Solver by FG"
18
+ description = "Sudoku Solver using MiniCPM-V-2 model by FG. Upload an image of a sudoku puzzle and ask a question to solve it."
19
 
20
  # Define the function for solving Sudoku
21
  def solve_sudoku(image, question):