update spaces.GPU

app.py

@@ -3,6 +3,7 @@ from threading import Thread
 from typing import Iterator, List, Tuple, Dict, Any
 
 import gradio as gr
+import spaces
 import torch
 from transformers import (
     TrainingArguments,
@@ -406,6 +407,7 @@ lora_config = LoraConfig(
 pretrained_model = get_peft_model(pretrained, lora_config)
 print(pretrained_model)
 
+@spaces.GPU(duration=30, queue=False)
 def run_training() -> str:
     """
     Hàm huấn luyện mô hình sử dụng GPU với thời gian hạn chế.
@@ -465,6 +467,7 @@ def run_training() -> str:
     return "Huấn luyện hoàn tất hoặc đã tiếp tục từ checkpoint."
 
 # Hàm Tự Động Hóa Việc Gọi Lặp Lại Hàm Huấn Luyện
+@spaces.GPU(duration=30, queue=False)
 def continuous_training(total_steps=300, steps_per_call=50):
     """
     Hàm tự động gọi lại `run_training` để hoàn thành quá trình huấn luyện.
@@ -536,6 +539,7 @@ def continuous_training(total_steps=300, steps_per_call=50):
 
 # ---------------------------- Giao Diện Gradio ---------------------------- #
 
+@spaces.GPU(duration=30, queue=False)
 def generate(
     message: str,
     chat_history: List[Tuple[str, str]],