prithivMLmods commited on
Commit
4699b28
·
verified ·
1 Parent(s): 0e5cdf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -128,7 +128,7 @@ def model_chat(prompt, image, is_batch=False):
128
  prompts = [f"<s>{p} <Answer/>" for p in prompts]
129
  prompt_inputs = processor.tokenizer(
130
  prompts,
131
- add_special_tokens=False,
132
  return_tensors="pt",
133
  padding=True
134
  ).to(device)
 
128
  prompts = [f"<s>{p} <Answer/>" for p in prompts]
129
  prompt_inputs = processor.tokenizer(
130
  prompts,
131
+ add_special_tokens=False, # Explicitly set to False
132
  return_tensors="pt",
133
  padding=True
134
  ).to(device)