ynhe commited on
Commit
f90cdc6
·
verified ·
1 Parent(s): 8377215

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -227,6 +227,7 @@ def gradio_ask(user_message, chatbot):
227
 
228
  @spaces.GPU
229
  def gradio_answer(chatbot, sys_prompt, user_prompt, video_tensor, chat_state, num_beams, temperature, do_sample=False):
 
230
  response, chat_state = model.chat(tokenizer,
231
  sys_prompt,
232
  user_prompt,
 
227
 
228
  @spaces.GPU
229
  def gradio_answer(chatbot, sys_prompt, user_prompt, video_tensor, chat_state, num_beams, temperature, do_sample=False):
230
+ video_tensor = video_tensor.to(model.device)
231
  response, chat_state = model.chat(tokenizer,
232
  sys_prompt,
233
  user_prompt,