Spaces:
Running
on
Zero
Running
on
Zero
Update src/app/response.py
Browse files- 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)
|