Jiaqi-hkust commited on
Commit
d47e225
·
verified ·
1 Parent(s): 974ec1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -132,7 +132,7 @@ def gradio_ask(user_message, chatbot, chat_state):
132
  return '', chatbot, chat_state
133
 
134
  @spaces.GPU
135
- def gradio_answer(chatbot, chat_state, img_list, num_beams, temperature):
136
  llm_message = chat.upload_video_without_audio(gr_video, chat_state, img_list)
137
  llm_message = chat.answer(conv=chat_state,
138
  img_list=img_list,
@@ -235,7 +235,7 @@ with gr.Blocks() as demo:
235
 
236
  start_time = time.time()
237
  text_input.submit(gradio_ask, [text_input, chatbot, chat_state], [text_input, chatbot, chat_state]).then(
238
- gradio_answer, [chatbot, chat_state, img_list, num_beams, temperature], [chatbot, chat_state, img_list]
239
  )
240
  end_time = time.time()
241
  print('Time:', end_time - start_time)
 
132
  return '', chatbot, chat_state
133
 
134
  @spaces.GPU
135
+ def gradio_answer(video, chatbot, chat_state, img_list, num_beams, temperature):
136
  llm_message = chat.upload_video_without_audio(gr_video, chat_state, img_list)
137
  llm_message = chat.answer(conv=chat_state,
138
  img_list=img_list,
 
235
 
236
  start_time = time.time()
237
  text_input.submit(gradio_ask, [text_input, chatbot, chat_state], [text_input, chatbot, chat_state]).then(
238
+ gradio_answer, [video, chatbot, chat_state, img_list, num_beams, temperature], [chatbot, chat_state, img_list]
239
  )
240
  end_time = time.time()
241
  print('Time:', end_time - start_time)