Update model.py

model.py

@@ -9,7 +9,10 @@ device = "cuda"
 
 # Load the model and tokenizer
 model = AutoModel.from_pretrained(
-    "openbmb/MiniCPM-V-2_6", trust_remote_code=True, attn_implementation='sdpa', torch_dtype=torch.bfloat16
+    "openbmb/MiniCPM-V-2_6",
+    trust_remote_code=True,
+    attn_implementation="sdpa",
+    torch_dtype=torch.bfloat16,
 )
 model = model.to(device=device)
 tokenizer = AutoTokenizer.from_pretrained(
@@ -22,11 +25,10 @@ model.eval()
 def answer_question(image, question):
     """
     Generates an answer to a given question based on the provided image and question.
-    
+
     Args:
         - image (str): The path to the image file.
         - question (str): The question text.
-
     Returns:
         str: The generated answer to the question.
     """