ynhe commited on
Commit
d7ac17f
1 Parent(s): 305df5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  try:
3
  token =os.environ['HF_TOKEN']
4
  except:
@@ -203,7 +204,7 @@ def gradio_reset(chat_state, img_list):
203
  img_list = None
204
  return None, gr.update(value=None, interactive=True), gr.update(placeholder='Please upload your video first', interactive=False),gr.update(value="Upload & Start Chat", interactive=True), chat_state, img_list
205
 
206
-
207
  def upload_img( gr_video, num_segments, hd_num, padding):
208
  img_list = []
209
  if gr_video is None:
@@ -225,7 +226,7 @@ def gradio_ask(user_message, chatbot):
225
  chatbot = chatbot + [[user_message, None]]
226
  return '', chatbot
227
 
228
-
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,
 
1
  import os
2
+ import spaces
3
  try:
4
  token =os.environ['HF_TOKEN']
5
  except:
 
204
  img_list = None
205
  return None, gr.update(value=None, interactive=True), gr.update(placeholder='Please upload your video first', interactive=False),gr.update(value="Upload & Start Chat", interactive=True), chat_state, img_list
206
 
207
+ @spaces.GPU
208
  def upload_img( gr_video, num_segments, hd_num, padding):
209
  img_list = []
210
  if gr_video is None:
 
226
  chatbot = chatbot + [[user_message, None]]
227
  return '', chatbot
228
 
229
+ @spaces.GPU
230
  def gradio_answer(chatbot, sys_prompt, user_prompt, video_tensor, chat_state, num_beams, temperature, do_sample=False):
231
  response, chat_state = model.chat(tokenizer,
232
  sys_prompt,