paralym commited on
Commit
7383108
1 Parent(s): 15c4e1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -452,10 +452,12 @@ def bot(history, temperature, top_p, max_output_tokens):
452
  print(input_ids.device)
453
  print(image_tensor.device)
454
 
 
 
455
  generate_kwargs = dict(
456
  inputs=input_ids,
457
  streamer=streamer,
458
- images=image_tensor,
459
  do_sample=True,
460
  temperature=temperature,
461
  top_p=top_p,
 
452
  print(input_ids.device)
453
  print(image_tensor.device)
454
 
455
+
456
+
457
  generate_kwargs = dict(
458
  inputs=input_ids,
459
  streamer=streamer,
460
+ images=[image_tensor] if is_video else image_tensor,
461
  do_sample=True,
462
  temperature=temperature,
463
  top_p=top_p,