Update app.py

app.py

@@ -11,28 +11,10 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, Auto
 from huggingface_hub import hf_hub_download
 import spaces 
 
-# ==========================================
-# 🚨 THE ALIBI ZERO-GPU BUGFIX PATCH 🚨
-# ==========================================
-# Hugging Face ZeroGPU uses a "meta" device to trace memory. 
-# DNABERT's custom ALiBi code hardcodes CPU tensors, causing a fatal collision.
-# We intercept PyTorch's arange function to dynamically align the tensors.
-_orig_arange = torch.arange
-def _alibi_safe_arange(*args, **kwargs):
-    tensor = _orig_arange(*args, **kwargs)
-    try:
-        # If ZeroGPU has forced the environment into 'meta' mode, we align the tensor
-        if torch.Tensor().device.type == 'meta' or getattr(torch, 'get_default_device', lambda: torch.device('cpu'))().type == 'meta':
-            return tensor.to('meta')
-    except:
-        pass
-    return tensor
-torch.arange = _alibi_safe_arange
-
 # ===================================
 # 1. LOAD AI MODELS (GLOBALLY CACHED)
 # ===================================
-print("Waking up the Genomic Oracle... Applying ALiBi Meta Patch.\n")
+print("Waking up the Genomic Oracle...\n")
 
 # A. Kadir's Gatekeeper 
 clf_coding = joblib.load("coding_classifier_universal.joblib")
@@ -64,7 +46,7 @@ FEATURE_DICT = {
 # ==============================================
 # 2. CORE INFERENCE ENGINE (ZeroGPU Accelerated)
 # ==============================================
-@spaces.GPU  # ZeroGPU automatically manages moving the global models to the A100!
+@spaces.GPU  # ZeroGPU natively manages the global models!
 def run_deep_learning_cascade(dna_sequence):
     device = torch.device("cuda")  
     clean_seq = "".join(dna_sequence.split()).upper()