Fix: robust HfFolder shim via sys.modules dict injection for gradio 5.0.0

app.py

@@ -12,10 +12,13 @@ import traceback
 from pathlib import Path
 
 # ── Compatibility shim: gradio 5.0.0 + huggingface_hub >= 0.27.0 ──
-# HfFolder was removed in huggingface_hub 0.27 but gradio 5.0.0 still imports it
+# HfFolder was removed in huggingface_hub 0.27+ but gradio 5.0.0 still imports it.
+# We inject a minimal stub directly into huggingface_hub's module dict BEFORE gradio loads.
 try:
+    import sys
     import huggingface_hub
-    if not hasattr(huggingface_hub, "HfFolder"):
+    hf_mod = sys.modules.get("huggingface_hub") or huggingface_hub
+    if "HfFolder" not in hf_mod.__dict__:
         class _HfFolderCompat:
             @staticmethod
             def get_token():
@@ -26,9 +29,15 @@ try:
             @staticmethod
             def delete_token():
                 pass
-        huggingface_hub.HfFolder = _HfFolderCompat()
-except Exception:
-    pass
+        # Inject into the real module dict so `from huggingface_hub import HfFolder` works
+        hf_mod.__dict__["HfFolder"] = _HfFolderCompat()
+        if hasattr(hf_mod, "__all__") and "HfFolder" not in hf_mod.__all__:
+            hf_mod.__all__.append("HfFolder")
+        print("[ViralImages] HfFolder compatibility shim applied.")
+    else:
+        print("[ViralImages] HfFolder already present, no shim needed.")
+except Exception as e:
+    print(f"[ViralImages] HfFolder shim failed (non-fatal): {e}")
 
 # Ensure our modules are importable
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))