Update app.py

app.py

@@ -9,12 +9,13 @@ device = "cpu"  # ✅ اجباراً فقط روی CPU اجرا شود
 def train_model(dataset_url, model_url, epochs):
     try:
         # 🚀 بارگیری مدل و توکنایزر با اجازه اجرای کد سفارشی و بدون نیاز به GPU
+        # 🚀 بارگیری مدل و توکنایزر بدون نیاز به GPU
         tokenizer = AutoTokenizer.from_pretrained(model_url, trust_remote_code=True)
         model = AutoModelForCausalLM.from_pretrained(
             model_url, 
             trust_remote_code=True, 
-            torch_dtype=torch.float32,  # ✅ تغییر به float32 برای CPU
-            device_map="cpu"  # ✅ اجباری کردن اجرای روی CPU
+            torch_dtype=torch.float32,  # ✅ اجرا روی CPU
+            device_map="cpu"
         )
 
         # ✅ تنظیم LoRA برای کاهش مصرف حافظه