Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,7 +51,12 @@ FEATURE_DICT = {
|
|
| 51 |
# ==============================================
|
| 52 |
@spaces.GPU
|
| 53 |
def run_deep_learning_cascade(dna_sequence):
|
| 54 |
-
device = torch.device("cuda")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
clean_seq = "".join(dna_sequence.split()).upper()
|
| 56 |
|
| 57 |
# --- LEVEL 1: Base Embedding & Kadir's Gatekeeper ---
|
|
|
|
| 51 |
# ==============================================
|
| 52 |
@spaces.GPU
|
| 53 |
def run_deep_learning_cascade(dna_sequence):
|
| 54 |
+
device = torch.device("cuda")
|
| 55 |
+
|
| 56 |
+
# 🚨 THE FINAL KEY: Teleport the CPU-locked models into the A100 GPU!
|
| 57 |
+
model_base.to(device)
|
| 58 |
+
model_promoter.to(device)
|
| 59 |
+
model_pheno.to(device)
|
| 60 |
clean_seq = "".join(dna_sequence.split()).upper()
|
| 61 |
|
| 62 |
# --- LEVEL 1: Base Embedding & Kadir's Gatekeeper ---
|