Spaces:
Sleeping
Sleeping
chore: Update app.py with improved Gradio interface for Sudoku Solver
Browse files
app.py
CHANGED
@@ -1,22 +1,23 @@
|
|
1 |
import gradio as gr
|
2 |
-
# Use a pipeline as a high-level helper
|
3 |
from transformers import pipeline
|
4 |
|
|
|
5 |
pipe = pipeline("visual-question-answering", model="openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True)
|
6 |
|
|
|
7 |
image = gr.Image(type="pil", label="Image")
|
8 |
-
question = "Using the
|
9 |
-
|
10 |
answer = gr.Textbox(label="Answer", show_label=True, show_copy_button=True)
|
11 |
|
12 |
-
|
13 |
title = "Sudoku Solver by FG"
|
14 |
description = "Sudoku Solver using MiniCPM-Llama3-V-2_5"
|
15 |
|
16 |
-
|
17 |
def solve_sudoku(image, question):
|
18 |
-
|
|
|
19 |
|
|
|
20 |
demo = gr.Interface(
|
21 |
fn=solve_sudoku,
|
22 |
inputs=[image, question],
|
@@ -24,6 +25,7 @@ demo = gr.Interface(
|
|
24 |
title=title,
|
25 |
description=description,
|
26 |
theme="compact",
|
27 |
-
)
|
28 |
|
29 |
-
|
|
|
|
1 |
import gradio as gr
|
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
+
# Initialize the pipeline
|
5 |
pipe = pipeline("visual-question-answering", model="openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True)
|
6 |
|
7 |
+
# Define the Gradio components
|
8 |
image = gr.Image(type="pil", label="Image")
|
9 |
+
question = gr.Textbox(value="Using the standard 9x9 sudoku format, solve the sudoku puzzle in the image correctly.", label="Question")
|
|
|
10 |
answer = gr.Textbox(label="Answer", show_label=True, show_copy_button=True)
|
11 |
|
|
|
12 |
title = "Sudoku Solver by FG"
|
13 |
description = "Sudoku Solver using MiniCPM-Llama3-V-2_5"
|
14 |
|
15 |
+
# Define the function for solving Sudoku
|
16 |
def solve_sudoku(image, question):
|
17 |
+
result = pipe(image, question)
|
18 |
+
return result[0]['answer']
|
19 |
|
20 |
+
# Create the Gradio interface
|
21 |
demo = gr.Interface(
|
22 |
fn=solve_sudoku,
|
23 |
inputs=[image, question],
|
|
|
25 |
title=title,
|
26 |
description=description,
|
27 |
theme="compact",
|
28 |
+
)
|
29 |
|
30 |
+
# Launch the interface
|
31 |
+
demo.launch(share=True)
|