vilarin commited on
Commit
d18aa37
1 Parent(s): 5cd56f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def stream_chat(message, history: list, system: str, temperature: float, max_new
63
  image = None
64
 
65
  prompt = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
66
- inputs = processor(prompt, images=[image], return_tensors="pt").to(0)
67
 
68
  generate_kwargs = dict(
69
  max_new_tokens=max_new_tokens,
 
63
  image = None
64
 
65
  prompt = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
66
+ inputs = processor(prompt, images=image, return_tensors="pt").to(0)
67
 
68
  generate_kwargs = dict(
69
  max_new_tokens=max_new_tokens,