sitammeur commited on
Commit
2d2be76
·
verified ·
1 Parent(s): 00fa929

Update src/app/response.py

Browse files
Files changed (1) hide show
  1. src/app/response.py +1 -1
src/app/response.py CHANGED
@@ -35,7 +35,7 @@ def caption_image(image: PIL.Image.Image, max_new_tokens: int, sampling: bool) -
35
  gr.Warning("Please provide an image.")
36
 
37
  # Prepare the inputs
38
- prompt = "caption en"
39
  model_inputs = (
40
  processor(text=prompt, images=image, return_tensors="pt")
41
  .to(torch.bfloat16)
 
35
  gr.Warning("Please provide an image.")
36
 
37
  # Prepare the inputs
38
+ prompt = "<image>caption en"
39
  model_inputs = (
40
  processor(text=prompt, images=image, return_tensors="pt")
41
  .to(torch.bfloat16)