hoduyquocbao commited on
Commit
b904b20
1 Parent(s): 502b366
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -275,7 +275,7 @@ def handle_functions(function_call: Dict[str, Any], prompt: str, chat_history: L
275
 
276
  # ---------------------------- Giao Diện Gradio ---------------------------- #
277
 
278
- @spaces.GPU(duration=60, queue=False)
279
  def generate(
280
  message: str,
281
  chat_history: List[Tuple[str, str]],
@@ -458,7 +458,7 @@ class SaveCheckpointCallback(TrainerCallback):
458
  return control # Trả về đối tượng control hiện tại
459
 
460
  # Định Nghĩa Hàm Huấn Luyện với Decorator @spaces.GPU
461
- @spaces.GPU(duration=60, queue=False)
462
  def run_training():
463
  """
464
  Hàm huấn luyện mô hình sử dụng GPU với thời gian hạn chế.
@@ -586,4 +586,4 @@ with gr.Blocks(css="""
586
  chat_interface.render() # Hiển thị giao diện trò chuyện
587
 
588
  if __name__ == "__main__":
589
- demo.queue(max_size=20).launch() # Khởi chạy ứng dụng Gradio với hàng đợi kích thước tối đa là 20
 
275
 
276
  # ---------------------------- Giao Diện Gradio ---------------------------- #
277
 
278
+ @spaces.GPU(duration=30, queue=False)
279
  def generate(
280
  message: str,
281
  chat_history: List[Tuple[str, str]],
 
458
  return control # Trả về đối tượng control hiện tại
459
 
460
  # Định Nghĩa Hàm Huấn Luyện với Decorator @spaces.GPU
461
+ @spaces.GPU(duration=30, queue=False)
462
  def run_training():
463
  """
464
  Hàm huấn luyện mô hình sử dụng GPU với thời gian hạn chế.
 
586
  chat_interface.render() # Hiển thị giao diện trò chuyện
587
 
588
  if __name__ == "__main__":
589
+ demo.queue(max_size=30).launch() # Khởi chạy ứng dụng Gradio với hàng đợi kích thước tối đa là 20