Update app.py

app.py

@@ -1,61 +1,32 @@
 import sys
 import subprocess
-import pkg_resources
-
-# List of required packages
-required_packages = {
-    "gradio": None,  # None means any version is acceptable
-    "torch": None,
-    "transformers": None,
-    "accelerate": None,
-    "pydantic": None,
-    "einops": None,
-    "timm": None,
-    "av": None
-}
-
-# Function to check if package is installed
-def is_package_installed(package_name):
-    try:
-        pkg_resources.get_distribution(package_name)
-        return True
-    except pkg_resources.DistributionNotFound:
-        return False
-
-# Install missing packages
-for package, version in required_packages.items():
-    if not is_package_installed(package):
-        print(f"Installing {package}...")
-        if version:
-            subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}=={version}"])
-        else:
-            subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+import os
 
-# Try to import OpenCV, install if missing
-try:
-    import cv2
-    print("OpenCV already installed.")
-except ImportError:
-    print("Installing OpenCV...")
+# First, try to install all dependencies
+packages_to_install = [
+    "gradio",
+    "torch",
+    "transformers", 
+    "accelerate",
+    "einops",
+    "timm",
+    "av",
+    "opencv-python-headless"  # Using headless version for better compatibility
+]
+
+for package in packages_to_install:
+    print(f"Installing {package}...")
     try:
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python-headless"])
-        import cv2
-        print("OpenCV successfully installed.")
-    except:
-        print("Failed to install opencv-python-headless, trying opencv-python...")
-        try:
-            subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python"])
-            import cv2
-            print("OpenCV successfully installed.")
-        except:
-            print("Failed to install OpenCV. Continuing without it...")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+        print(f"Successfully installed {package}")
+    except Exception as e:
+        print(f"Error installing {package}: {e}")
 
-# Now import the rest of the packages
+# Now proceed with the actual application
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gc
-import os
 import datetime
 import time
 import spaces
@@ -63,7 +34,6 @@ import spaces
 # --- 설정 ---
 MODEL_ID = "naver-hyperclovax/HyperCLOVAX-SEED-Vision-Instruct-3B"
 MAX_NEW_TOKENS = 512
-CPU_THREAD_COUNT = 4 # 필요시 조절
 
 # Hugging Face 토큰 설정 - 환경 변수에서 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -74,7 +44,6 @@ if not HF_TOKEN:
 print("--- 환경 설정 ---")
 print(f"PyTorch 버전: {torch.__version__}")
 print(f"실행 장치: {torch.device('cuda' if torch.cuda.is_available() else 'cpu')}")
-print(f"Torch 스레드: {torch.get_num_threads()}")
 print(f"HF_TOKEN 설정 여부: {'있음' if HF_TOKEN else '없음'}")
 
 # --- 모델 및 토크나이저 로딩 ---