diff --git a/__init__.py b/__init__.py
index efca5a9..fffb2ea 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 efficientnet_pytorch
-except ModuleNotFoundError:
-    os.system('pip install efficientnet_pytorch')
+# # For requirements.
+# try:
+#     import efficientnet_pytorch
+# except ModuleNotFoundError:
+#     os.system('pip install efficientnet_pytorch')
diff --git a/pytorch/__init__.py b/pytorch/__init__.py
index 37f5bd7..ed43fad 100644
--- a/pytorch/__init__.py
+++ b/pytorch/__init__.py
@@ -11,3 +11,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+
+import os
+
+# For requirements.
+try:
+    import efficientnet_pytorch
+except ModuleNotFoundError:
+    os.system('pip install efficientnet_pytorch')
\ No newline at end of file