aarbelle commited on
Commit
6ee40ff
·
verified ·
1 Parent(s): 7b82a16

Update src/app.py

Browse files

The image token for granite vision is `<image>` not `[Image]`

Files changed (1) hide show
  1. src/app.py +1 -1
src/app.py CHANGED
@@ -123,7 +123,7 @@ def get_text_from_content(content):
123
  if item["type"] == "text":
124
  texts.append(item["text"])
125
  elif item["type"] == "image":
126
- texts.append("[Image]")
127
  return " ".join(texts)
128
 
129
  @spaces.GPU
 
123
  if item["type"] == "text":
124
  texts.append(item["text"])
125
  elif item["type"] == "image":
126
+ texts.append("<image>")
127
  return " ".join(texts)
128
 
129
  @spaces.GPU