From daa5d12fcd8e4631bb5741f820ed7fd6e3b7e9cc Mon Sep 17 00:00:00 2001
From: WangZiFan <musknaivew0911@gmail.com>
Date: Fri, 27 Feb 2026 11:37:39 +0800
Subject: [PATCH] Update model/__init__.py

---
 model/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/model/__init__.py b/model/__init__.py
index 8d22e5a..4692f80 100644
--- a/model/__init__.py
+++ b/model/__init__.py
@@ -1 +1,2 @@
-from .modelling import TransClassifier
\ No newline at end of file
+from .modelling import TransClassifier
+from .focal_loss import FocalLoss
\ No newline at end of file