Update src/demo/gradio_utils.py

src/demo/gradio_utils.py

@@ -1,29 +1,22 @@
-import sys
 import spaces
 import torch
-import importlib
 import gradio as gr
 import numpy as np
 
 # Walk around container build-up
-package_name = "huggingface_hub"
-required_version = "0.25.1"
-try:
-    huggingface_hub = importlib.import_module(package_name)
-    if huggingface_hub.__version__ != required_version:
-        raise ImportError
-except ImportError:
+def get_installed_version(package="huggingface_hub"):
+    import importlib
+    try:
+        module = importlib.import_module(package)
+        return module.__version__
+    except ImportError:
+        return None
+pkg_version = get_installed_version("huggingface_hub")
+if pkg_version != "0.25.1":
+    import sys
     import subprocess
-    subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package_name}=={required_version}"])
-
-# Remove old version from memory
-if package_name in sys.modules:
-    del sys.modules[package_name]
-
-importlib.invalidate_caches()
-huggingface_hub = importlib.import_module(package_name)
-importlib.reload(huggingface_hub)
-print(f"Now the `huggingface_hub` version is {huggingface_hub.__version__}")
+    print(f"Installing huggingface_hub==0.25.1 (Current version: {pkg_version})")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", f"huggingface_hub==0.25.1"])
     
 from src.demo.utils import (
     create_model, get_spec_pil, get_mask_region, get_mask_regions, func_clear,