diff --git a/__init__.py b/__init__.py
index 9379ed6..8516979 100644
--- a/__init__.py
+++ b/__init__.py
@@ -12,10 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# import os
+import os
 
-# # For requirements.
-# try:
-#     import timm
-# except ModuleNotFoundError:
-#     os.system('pip install timm')
+# For requirements.
+try:
+    import timm
+except ModuleNotFoundError:
+    os.system('pip install timm')