Spaces:
Sleeping
Sleeping
Update app.py with improved Gradio interface and update model description
Browse files
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-
|
| 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-
|
| 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):
|