Fix ZeroGPU

model/model.py

@@ -17,7 +17,6 @@ import torch
 from typing import *
 from rdkit import RDLogger
 RDLogger.DisableLog("rdApp.*")
-import spaces
 
 from xgboost import XGBClassifier, DMatrix
 
@@ -68,7 +67,6 @@ class DTIModel:
         morgan = [self._encode_smiles(s, radius, bits, features) for s in smiles]
         return np.array(morgan)
     
-    @spaces.GPU
     def _encode_sequence(self, sequence: str):
         # Clear torch cache
         torch.cuda.empty_cache()
@@ -88,12 +86,10 @@ class DTIModel:
                 print(e)
                 return None
 
-    @spaces.GPU
     def _encode_sequence_mult(self, sequences: List[str]):
         seq = [self._encode_sequence(sequence) for sequence in sequences]
         return np.array(seq)
 
-    @spaces.GPU
     def __predict_pair(self, drug_emb: np.ndarray, target_emb: np.ndarray, pred_leaf: bool):
         if drug_emb.shape[0] < target_emb.shape[0]:
             drug_emb = np.tile(drug_emb, (len(target_emb), 1))