Update app.py

app.py

@@ -1,20 +1,27 @@
 import os
-
-# ── Must be set BEFORE importing transformers/huggingface_hub ──────────────────
-# Suppresses the interactive "Do you wish to run the custom code? [y/N]" prompt.
-os.environ["TRUST_REMOTE_CODE"]          = "1"
+import builtins
+
+# ── MUST happen before ANY other import ───────────────────────────────────────
+# transformers calls builtins.input() for the "Do you wish to run the custom
+# code? [y/N]" prompt.  Patch it to always answer "y" silently.
+_real_input = builtins.input
+def _auto_yes(prompt=""):
+    if "custom code" in str(prompt).lower() or "trust" in str(prompt).lower():
+        return "y"
+    return _real_input(prompt)
+builtins.input = _auto_yes
+
+os.environ["TRUST_REMOTE_CODE"]            = "1"
 os.environ["HF_HUB_DISABLE_PROGRESS_BARS"] = "1"
-os.environ["TOKENIZERS_PARALLELISM"]     = "false"
-# Equivalent to answering "y" at the stdin prompt in non-TTY environments
-os.environ["HF_HUB_VERBOSITY"]          = "error"
+os.environ["TOKENIZERS_PARALLELISM"]       = "false"
+os.environ["HF_HUB_VERBOSITY"]            = "error"
 
 import streamlit as st
 import numpy as np
 import re
 from transformers import AutoModel
-from transformers.dynamic_module_utils import resolve_trust_remote_code  # noqa: F401
 
-# Monkey-patch: always return True so no prompt is shown anywhere
+# Belt-and-suspenders: patch the internal resolver too, after import
 try:
     import transformers.dynamic_module_utils as _dmu
     _dmu.resolve_trust_remote_code = lambda *a, **kw: True  # type: ignore