Faustrix commited on
Commit
b5e31dc
1 Parent(s): dd2411e

chore: Update app.py with improved Gradio interface for Sudoku Solver

Browse files
Files changed (1) hide show
  1. app.py +10 -8
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 standar 9x9 sudoku format, solve the the sudoku puzzle in the image correctly."
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
- return pipe(image, question)
 
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
- demo.launch(share=True)
 
 
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)