Update README.md

README.md

@@ -21,12 +21,54 @@ license: creativeml-openrail-m
 # Castor-3D-Portrait-Flux-LoRA
 
 <Gallery />
+**The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.**
 
+## Model description 
 
-## Trigger words
+**prithivMLmods/Castor-3D-Portrait-Flux-LoRAA**
 
-You should use `3D Portrait` to trigger the image generation.
+Image Processing Parameters 
 
+| Parameter                 | Value  | Parameter                 | Value  |
+|---------------------------|--------|---------------------------|--------|
+| LR Scheduler              | constant | Noise Offset              | 0.03   |
+| Optimizer                 | AdamW  | Multires Noise Discount   | 0.1    |
+| Network Dim               | 64     | Multires Noise Iterations | 10     |
+| Network Alpha             | 32     | Repeat & Steps           | 20 & 2.5K |
+| Epoch                     | 15     | Save Every N Epochs       | 1      |
+
+    Labeling: florence2-en(natural language & English)
+    
+    Total Images Used for Training : 19
+
+## Setting Up
+```
+import torch
+from pipelines import DiffusionPipeline
+
+base_model = "black-forest-labs/FLUX.1-dev"
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
+
+lora_repo = "prithivMLmods/Castor-3D-Portrait-Flux-LoRA"
+trigger_word = "3D Portrait"  # Leave trigger_word blank if not used.
+pipe.load_lora_weights(lora_repo)
+
+device = torch.device("cuda")
+pipe.to(device)
+```
+## App File Structure
+
+/project-root/
+
+├── .gitattributes         
+├── README.md      
+├── app.py             
+├── pythonproject.py     
+
+## Trigger words 🧨
+
+> [!WARNING]
+> **Trigger words:**  You should use `3D Portrait` to trigger the image generation.
 
 ## Download model