Update app.py with improved Gradio interface and change model's torch_dtype to float32

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import torch
-from PIL import Image
 from transformers import AutoModel, AutoTokenizer
 
 device = "cpu" 
@@ -24,16 +23,11 @@ description = "Sudoku Solver using MiniCPM-Llama3-V-2_5"
 
 # Define the function for solving Sudoku
 def solve_sudoku(image, question):
-    # Convert image to RGB format if not already in RGB
-    image = image.convert("RGB") if image.mode != 'RGB' else image
     msgs = [{"role": "user", "content": question}]
     res = model.chat(
         image=image,
         msgs=msgs,
         tokenizer=tokenizer,
-        sampling=False,
-        temperature=0.7,
-        stream=False,  # Enable streaming
         system_prompt="You are an expert in solving sudoku puzzles. Please solve the sudoku puzzle in the image correctly.",
     )
     return "".join(res)