For Check

app.py

@@ -11,7 +11,7 @@ import uvicorn
 import time
 import numpy as np
 import subprocess
-import pkg_resources
+import importlib.metadata
 
 # 모듈 설치 명령
 subprocess.run(["pip", "install", "transformers"], check=True)
@@ -24,8 +24,11 @@ subprocess.run(["pip", "install", "numpy==1.26.0"], check=True)
 modules = ['transformers', 'accelerate', 'peft', 'FlagEmbedding']
 
 for module in modules:
-    version = pkg_resources.get_distribution(module).version
-    print(f"{module} version: {version}")
+    try:
+        version = importlib.metadata.version(module)
+        print(f"{module} version: {version}")
+    except importlib.metadata.PackageNotFoundError:
+        print(f"{module} is not installed")
 
 #fmt: off
 from transformers import AutoModelForCausalLM, AutoTokenizer