fix error gradio

app.py

@@ -42,6 +42,9 @@ MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "128000"))  #
 # Xác định thiết bị sử dụng (GPU nếu có, ngược lại CPU)
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
+# Hiển thị sử dụng GPU hoặc CPU
+print(device)
+
 model_id = "meta-llama/Llama-3.2-3B-Instruct"  # ID mô hình
 
 # Tải tokenizer
@@ -421,7 +424,7 @@ def get_step_done() -> int:
 
 # ---------------------------- Định Nghĩa Huấn Luyện ---------------------------- #
 
-@gradio.GPU  # Sử dụng decorator phù hợp nếu cần
+@spaces.GPU  # Sử dụng decorator phù hợp nếu cần
 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ế.
@@ -481,7 +484,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
-@gradio.GPU
+@spaces.GPU
 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.