vilarin commited on
Commit
3090ac5
1 Parent(s): 6904764

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -68,6 +68,7 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
68
  inputs_ids = processor(inputs, image, return_tensors="pt").to(0)
69
 
70
  generate_kwargs = dict(
 
71
  max_new_tokens=max_new_tokens,
72
  temperature=temperature,
73
  do_sample=True,
 
68
  inputs_ids = processor(inputs, image, return_tensors="pt").to(0)
69
 
70
  generate_kwargs = dict(
71
+ streamer=streamer,
72
  max_new_tokens=max_new_tokens,
73
  temperature=temperature,
74
  do_sample=True,