Update README.md

README.md

@@ -33,7 +33,26 @@ Available on **PyPI**: [torchlosses](https://pypi.org/project/torchlosses/)
 
 Naga Adithya Kaushik (GenAIDevTOProd)
 
-## 📦 Installation
+## Installation
 
 ```bash
-pip install torchlosses
+pip install torchlosses
+
+## Usage
+
+import torch
+from torchlosses import FocalLoss, DiceLoss
+
+# Focal Loss for classification
+inputs = torch.randn(4, 5, requires_grad=True)  # logits for 5 classes
+targets = torch.randint(0, 5, (4,))
+criterion = FocalLoss()
+loss = criterion(inputs, targets)
+print("Focal Loss:", loss.item())
+
+# Dice Loss for segmentation
+inputs = torch.randn(4, 1, 8, 8)
+targets = torch.randint(0, 2, (4, 1, 8, 8))
+criterion = DiceLoss()
+loss = criterion(inputs, targets)
+print("Dice Loss:", loss.item())