Geonomic commited on
Commit
dcfc166
·
verified ·
1 Parent(s): 1fb66e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -36,18 +36,18 @@ def run_deep_learning_cascade(dna_sequence):
36
  clf_coding = joblib.load("coding_classifier_universal.joblib")
37
 
38
  tokenizer_base = AutoTokenizer.from_pretrained("DNABERT_Local", trust_remote_code=True)
39
- model_base = AutoModel.from_pretrained("DNABERT_Local", trust_remote_code=True).to(device)
40
  model_base.eval()
41
 
42
  tokenizer_promoter = AutoTokenizer.from_pretrained("llm_promoter_classifier_v2", trust_remote_code=True)
43
- model_promoter = AutoModelForSequenceClassification.from_pretrained("llm_promoter_classifier_v2", trust_remote_code=True).to(device)
44
  model_promoter.eval()
45
 
46
  lgbm_path = hf_hub_download(repo_id="Geonomic/Genomic-Oracle-Weights", filename="dnabert_lightgbm_model_feature_type.pkl")
47
  lightgbm_model = joblib.load(lgbm_path)
48
 
49
  tokenizer_pheno = AutoTokenizer.from_pretrained("Geonomic/Genomic-Oracle-Weights", trust_remote_code=True)
50
- model_pheno = AutoModelForSequenceClassification.from_pretrained("Geonomic/Genomic-Oracle-Weights", trust_remote_code=True).to(device)
51
  model_pheno.eval()
52
 
53
  # --- LEVEL 1: Base Embedding & Kadir's Gatekeeper ---
 
36
  clf_coding = joblib.load("coding_classifier_universal.joblib")
37
 
38
  tokenizer_base = AutoTokenizer.from_pretrained("DNABERT_Local", trust_remote_code=True)
39
+ model_base = AutoModel.from_pretrained("DNABERT_Local", trust_remote_code=True, low_cpu_mem_usage=False).to(device)
40
  model_base.eval()
41
 
42
  tokenizer_promoter = AutoTokenizer.from_pretrained("llm_promoter_classifier_v2", trust_remote_code=True)
43
+ model_promoter = AutoModelForSequenceClassification.from_pretrained("llm_promoter_classifier_v2", trust_remote_code=True, low_cpu_mem_usage=False).to(device)
44
  model_promoter.eval()
45
 
46
  lgbm_path = hf_hub_download(repo_id="Geonomic/Genomic-Oracle-Weights", filename="dnabert_lightgbm_model_feature_type.pkl")
47
  lightgbm_model = joblib.load(lgbm_path)
48
 
49
  tokenizer_pheno = AutoTokenizer.from_pretrained("Geonomic/Genomic-Oracle-Weights", trust_remote_code=True)
50
+ model_pheno = AutoModelForSequenceClassification.from_pretrained("Geonomic/Genomic-Oracle-Weights", trust_remote_code=True, low_cpu_mem_usage=False).to(device)
51
  model_pheno.eval()
52
 
53
  # --- LEVEL 1: Base Embedding & Kadir's Gatekeeper ---