Update app.py

app.py

@@ -6,11 +6,31 @@ import random
 import torch
 import spaces
 import requests
+import subprocess
+import importlib.util
 import gradio as gr
 from PIL import Image
 from threading import Thread
 from typing import Iterable, Optional, Tuple, List
 
+def check_and_install_package(package_name, import_name=None, pip_name=None):
+    """Check if a package is installed, and if not, install it."""
+    if import_name is None:
+        import_name = package_name
+    if pip_name is None:
+        pip_name = package_name
+
+    spec = importlib.util.find_spec(import_name)
+    if spec is None:
+        print(f"Installing {package_name}...")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", pip_name, "-q"])
+        print(f"✓ {package_name} installed successfully")
+    return True
+
+print("Checking and installing transformers==4.57.3 ...")
+check_and_install_package("transformers", "transformers", "transformers==4.57.3")
+print("Done!")
+
 from transformers import (
     Qwen2_5_VLForConditionalGeneration,
     AutoProcessor,