Update app.py

app.py

@@ -14,19 +14,22 @@ import spaces  # REQUIRED FOR ZEROGPU
 # ===================================
 # 1. LOAD AI MODELS (GLOBALLY CACHED)
 # ===================================
-print("Waking up the Genomic Oracle... Loading models into VRAM.")
+print("Waking up the Genomic Oracle...\n")
 
 # A. Kadir's Gatekeeper (Local in the Space)
+print("Loading Logression Model...\n")
 clf_coding = joblib.load("coding_classifier_universal.joblib")
 
 # B. Base DNABERT (Local in the Space)
+print("Loading foundational DNABERT Model Architecture...\n")
 tokenizer_base = AutoTokenizer.from_pretrained("DNABERT_Local", trust_remote_code=True)
-model_base = AutoModel.from_pretrained("DNABERT_Local", trust_remote_code=True)
+model_base = AutoModel.from_pretrained("DNABERT_Local", trust_remote_code=True, low_cpu_mem_usage=False)
 model_base.eval()
 
 # C. DNABERT-2 Promoter Model (Local in the Space)
+print("Loading DNABERT-2 Neural Network...\n")
 tokenizer_promoter = AutoTokenizer.from_pretrained("llm_promoter_classifier_v2", trust_remote_code=True)
-model_promoter = AutoModelForSequenceClassification.from_pretrained("llm_promoter_classifier_v2", trust_remote_code=True)
+model_promoter = AutoModelForSequenceClassification.from_pretrained("llm_promoter_classifier_v2", trust_remote_code=True, low_cpu_mem_usage=False)
 model_promoter.eval()
 
 # D. Multi-Feature LightGBM (Fetched from Cloud Repo)
@@ -35,9 +38,9 @@ lgbm_path = hf_hub_download(repo_id="Geonomic/Genomic-Oracle-Weights", filename=
 lightgbm_model = joblib.load(lgbm_path)
 
 # E. Custom ALiBi Lean/Obese BERT (Fetched from Cloud Repo)
-print("Downloading Phenotype BERT from Model Repo...")
+print("Downloading Phenotype BERT from Model Repository...")
 tokenizer_pheno = AutoTokenizer.from_pretrained("Geonomic/Genomic-Oracle-Weights", trust_remote_code=True)
-model_pheno = AutoModelForSequenceClassification.from_pretrained("Geonomic/Genomic-Oracle-Weights", trust_remote_code=True)
+model_pheno = AutoModelForSequenceClassification.from_pretrained("Geonomic/Genomic-Oracle-Weights", trust_remote_code=True, low_cpu_mem_usage=False)
 model_pheno.eval()
 
 # Structural Feature Dictionary
@@ -190,7 +193,7 @@ def gradio_inference(dna_sequence, run_mapping):
                 f"🎯 **Location:** {context['location']}",
                 f"🧬 **Strand:** {context['strand']}",
                 f"🧭 **Coordinates:** {context['start']:,} – {context['end']:,}",
-                f"🔬**Notes:** {context['metadata']}"
+                f"🔬 **Notes:** {context['metadata']}"
             ]
             context_output = "\n".join(context_lines)
         else: