Update app.py

app.py

@@ -1,8 +1,9 @@
+pip install --upgrade diffusers
 import os
 import gradio as gr
 import torch
-from diffusers import StableDiffusionPipeline, DreamBoothTrainer, TrainingArguments, Trainer
-from transformers import CLIPTextModel, CLIPTokenizer
+from diffusers import StableDiffusionPipeline
+from transformers import CLIPTextModel, CLIPTokenizer, Trainer, TrainingArguments
 from datasets import load_dataset
 import json
 import time
@@ -54,13 +55,13 @@ def train_model(user_email, images, progress):
     tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch16")
     text_encoder = CLIPTextModel.from_pretrained("openai/clip-vit-base-patch16")
 
-    # إعداد الـ Trainer باستخدام DreamBooth
-    trainer = DreamBoothTrainer(
+    # إعداد الـ Trainer باستخدام Trainer من Hugging Face
+    trainer = Trainer(
         model=pipe.unet, 
         args=training_args, 
         train_dataset=dataset["train"],
         tokenizer=tokenizer,
-        text_encoder=text_encoder,
+        data_collator=None,
     )
 
     # بدء عملية التدريب