Spaces:
Running
on
Zero
Running
on
Zero
hoduyquocbao
commited on
Commit
•
b904b20
1
Parent(s):
502b366
edit time
Browse files
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=
|
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=
|
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=
|
|
|
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
|