Update README.md

README.md

@@ -82,7 +82,12 @@ conv.append_message(conv.roles[1], None)
 prompt_question = conv.get_prompt()
 
 # - Create model inputs
-input_ids = tokenizer_image_token(prompt_question, tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt").unsqueeze(0).to(model.device)
+input_ids = tokenizer_image_token(
+  prompt_question,
+  tokenizer,
+  IMAGE_TOKEN_INDEX,
+  return_tensors="pt"
+).unsqueeze(0).to(model.device)
 image_sizes = [image.size]
 
 # - Generate model response
@@ -99,7 +104,11 @@ output = model.generate(
   temperature=temperature if do_sample else None,
   max_new_tokens=max_new_tokens,
 )
-output_parsed= tokenizer.decode(output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
+output_parsed= tokenizer.decode(
+  output[0],
+  skip_special_tokens=True,
+  clean_up_tokenization_spaces=False
+)
 	
 # - Process response as you wish ...
 #response= output_parsed.strip("\n").strip()