Update app.py

app.py

@@ -12,14 +12,25 @@ import re, torch, gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from huggingface_hub import hf_hub_download
 import spaces
-import os                                  # NEW
-
-# ──────────────────  hot-patch torch.compile  ────────────────────────────
-if hasattr(torch, "compile"):              # NEW
-    def _no_compile(model, *args, **kw):   # NEW
-        return model                       # NEW
-    torch.compile = _no_compile            # NEW
-    os.environ["TORCHINDUCTOR_DISABLED"] = "1"  # NEW
+import os, types                 # add `types`
+
+# ──────────────────  robust torch.compile shim  ─────────────────────────
+if hasattr(torch, "compile"):
+    def _no_compile(model: types.Any = None, *args, **kwargs):
+        """
+        1. If called as torch.compile(model, …)  → just return the model.
+        2. If called as torch.compile(**kw)      → return a decorator that
+           immediately gives back the class / fn it decorates.
+        """
+        if callable(model):      # pattern 1
+            return model
+        # pattern 2  (used by ModernBERT via @torch.compile(...))
+        def decorator(fn):
+            return fn
+        return decorator
+
+    torch.compile = _no_compile          # monkey-patch
+    os.environ["TORCHINDUCTOR_DISABLED"] = "1"
 
 # (everything below is unchanged)
 DEVICE        = torch.device("cuda" if torch.cuda.is_available() else "cpu")