Update app.py

app.py

@@ -408,7 +408,7 @@ def example_trigger(text_input, image, upload_flag, replace_flag, img_list):
 
     return upload_flag, replace_flag
 
-
+@spaces.GPU(duration=120)
 def gradio_ask(user_message, chatbot, chat_state, gr_img, img_list, upload_flag, replace_flag):
     if len(user_message) == 0:
         text_box_show = 'Input should not be empty!'
@@ -451,7 +451,7 @@ def gradio_ask(user_message, chatbot, chat_state, gr_img, img_list, upload_flag,
 
     return text_box_show, chatbot, chat_state, img_list, upload_flag, replace_flag
 
-
+@spaces.GPU(duration=120)
 def gradio_answer(chatbot, chat_state, img_list, temperature):
     llm_message = chat.answer(conv=chat_state,
                               img_list=img_list,
@@ -461,7 +461,7 @@ def gradio_answer(chatbot, chat_state, img_list, temperature):
     chatbot[-1][1] = llm_message
     return chatbot, chat_state
 
-
+@spaces.GPU(duration=120)
 def gradio_stream_answer(chatbot, chat_state, img_list, temperature):
     if len(img_list) > 0:
         if not isinstance(img_list[0], torch.Tensor):
@@ -480,7 +480,7 @@ def gradio_stream_answer(chatbot, chat_state, img_list, temperature):
     chat_state.messages[-1][1] = '</s>'
     return chatbot, chat_state
 
-
+@spaces.GPU(duration=120)
 def gradio_visualize(chatbot, gr_img):
     if isinstance(gr_img, dict):
         gr_img, mask = gr_img['image'], gr_img['mask']