SodokuSolver / app.py
Faustrix's picture
feat: Update app.py with compact theme for interface
906cc8b
raw
history blame
780 Bytes
import gradio as gr
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("visual-question-answering", model="openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True)
image = gr.image(type="pil", label="Image")
question = "Using the standar 9x9 sudoku format, solve the the sudoku puzzle in the image correctly."
answer = gr.outputs.Textbox(label="Answer", show_label=True, show_copy_button=True)
title = "Sudoku Solver by FG"
description = "Sudoku Solver using MiniCPM-Llama3-V-2_5"
def solve_sudoku(image, question):
return pipe(image, question)
demo = gr.Interface(
fn=solve_sudoku,
inputs=[image, question],
outputs=answer,
title=title,
description=description,
theme="compact",
)
demo.launch(share=True)