Update app.py

app.py

@@ -1,4 +1,11 @@
 import os
+# --- Prevent SIGSEGV (exit 139) from TensorFlow + PyTorch native lib clashes ---
+# TF and torch each bundle their own OpenMP/MKL; loaded together they can collide
+# and crash at the C level. These settings make them coexist and reduce memory.
+os.environ.setdefault("KMP_DUPLICATE_LIB_OK", "TRUE")
+os.environ.setdefault("OMP_NUM_THREADS", "1")
+os.environ.setdefault("MKL_NUM_THREADS", "1")
+os.environ.setdefault("OPENBLAS_NUM_THREADS", "1")
 # Quiet TensorFlow logs (must be set before importing tensorflow)
 os.environ.setdefault("TF_CPP_MIN_LOG_LEVEL", "3")
 os.environ.setdefault("TF_ENABLE_ONEDNN_OPTS", "0")
@@ -36,6 +43,10 @@ def get_protbert():
     if _protbert_model is None:
         import torch
         from transformers import BertTokenizer, BertModel
+        try:
+            torch.set_num_threads(1)  # reduce native threading conflicts with TF
+        except Exception:
+            pass
         _torch = torch
         _device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         _protbert_tokenizer = BertTokenizer.from_pretrained(