NeelakshSaxena commited on
Commit
cd7aae8
Β·
1 Parent(s): 4d14417

Fix tiny prompt content format

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -312,10 +312,7 @@ def build_prompt(processor_obj, user_text: str) -> str:
312
  messages = [
313
  {
314
  "role": "user",
315
- "content": [
316
- {"type": "image"},
317
- {"type": "text", "text": user_text},
318
- ],
319
  }
320
  ]
321
  tokenizer = getattr(processor_obj, "tokenizer", None)
 
312
  messages = [
313
  {
314
  "role": "user",
315
+ "content": f"<image>\n{user_text}",
 
 
 
316
  }
317
  ]
318
  tokenizer = getattr(processor_obj, "tokenizer", None)