Update app.py

app.py

@@ -107,7 +107,7 @@ class LoRAImageTrainer:
             target_modules=target_modules,  
             lora_dropout=lora_dropout,  
             bias="none",  
-            task_type=TaskType.DIFFUSION,  
+            task_type=TaskType.CAUSAL_LM,  # ✅ CORREÇÃO PRINCIPAL: DIFFUSION → CAUSAL_LM
         )  
   
     def prepare_image_dataset(self, image_files: List[str], captions: List[str], resolution: int = 512) -> List[Dict]:  
@@ -284,7 +284,7 @@ class LoRAImageTrainer:
                 "target_modules": lora_config.target_modules,
                 "lora_dropout": lora_dropout,
                 "bias": "none",
-                "task_type": "DIFFUSION",
+                "task_type": "CAUSAL_LM",  # ✅ CORREÇÃO AQUI TAMBÉM: DIFFUSION → CAUSAL_LM
                 "base_model_name": model_name,
                 "training_info": {
                     "num_epochs": num_epochs,