Update app.py

app.py

@@ -22,8 +22,7 @@ from torch.utils.data import DataLoader
 from pathlib import Path
 from safetensors.torch import load_file
 from transformers import AutoTokenizer, AutoModelForCausalLM, PretrainedConfig, PreTrainedModel
-import gradio as gr
-from gradio import spaces # <-- ADD THIS IMPORT
+import gradio as gr# <-- ADD THIS IMPORT
 import os
 from datetime import datetime
 import threading
@@ -31,7 +30,7 @@ import time
 import traceback
 
 # --- RLHF & Training Imports ---
-from huggingface_hub import HfApi, login
+from huggingface_hub import HfApi, login, spaces
 from datasets import Dataset, load_dataset, concatenate_datasets
 from peft import LoraConfig, get_peft_model