Update app.py

app.py

@@ -40,6 +40,8 @@ def create_captions_rich(image: Image.Image) -> str:
     prompt = "caption en"
     # 이미지 데이터를 전처리하여 processor에 전달
     image_tensor = processor(images=image, return_tensors="pt").pixel_values.to("cpu")
+    # 이미지 범위 조정 [0, 1]에서 [0, 255]로
+    image_tensor = (image_tensor * 255).type(torch.uint8)
     model_inputs = processor(text=prompt, images=image_tensor, return_tensors="pt").to("cpu")
     input_len = model_inputs["input_ids"].shape[-1]