pan-yl commited on
Commit
b4de814
1 Parent(s): bf225fc
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -427,7 +427,7 @@ class ChatBotUI(object):
427
  def set_callbacks(self, *args, **kwargs):
428
 
429
  ########################################
430
- @spaces.GPU(duration=120)
431
  def change_model(model_name):
432
  if model_name not in self.model_choices:
433
  gr.Info('The provided model name is not a valid choice!')
@@ -577,7 +577,7 @@ class ChatBotUI(object):
577
  outputs=[self.history, self.chatbot, self.text, self.gallery])
578
 
579
  ########################################
580
- @spaces.GPU(duration=120)
581
  def run_chat(message,
582
  extend_prompt,
583
  history,
 
427
  def set_callbacks(self, *args, **kwargs):
428
 
429
  ########################################
430
+ @spaces.GPU(duration=600)
431
  def change_model(model_name):
432
  if model_name not in self.model_choices:
433
  gr.Info('The provided model name is not a valid choice!')
 
577
  outputs=[self.history, self.chatbot, self.text, self.gallery])
578
 
579
  ########################################
580
+ @spaces.GPU(duration=600)
581
  def run_chat(message,
582
  extend_prompt,
583
  history,