Update src/paligemma/response.py

src/paligemma/response.py

@@ -39,13 +39,11 @@ def caption_image(image: PIL.Image.Image, max_new_tokens: int, language: str) ->
     try:
         # Check if image is None
         if not image:
-            gr.Warning("Please provide an image.")
+            gr.Error("Please provide an image.")
 
         # Prepare the inputs
         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)