Update src/paligemma/response.py

src/paligemma/response.py

@@ -45,6 +45,7 @@ def caption_image(image: PIL.Image.Image, max_new_tokens: int, language: str) ->
         language = language_dict[language]
         print(language)
         prompt = f"<image>caption {language}"
+        print(prompt)
         model_inputs = (
             processor(text=prompt, images=image, return_tensors="pt")
             .to(torch.bfloat16)