han1997 commited on
Commit
6b6fe1e
·
verified ·
1 Parent(s): 966597c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,8 +65,8 @@ def bot_streaming(message, history, temperature, top_k, max_new_tokens):
65
  prompt_text = prompt_builder.get_prompt()
66
 
67
  generation_kwargs = {
68
- "pixel_values": image,
69
- "input_ids": prompt_text,
70
  "streamer": streamer,
71
  "max_new_tokens": max_new_tokens,
72
  "use_cache": True,
 
65
  prompt_text = prompt_builder.get_prompt()
66
 
67
  generation_kwargs = {
68
+ "image": image,
69
+ "prompt_text": prompt_text,
70
  "streamer": streamer,
71
  "max_new_tokens": max_new_tokens,
72
  "use_cache": True,