Upload 5 files

app.py

@@ -3,7 +3,7 @@ import plotly.graph_objects as go
 import numpy as np
 import pandas as pd
 from model.model import DTIModel
-
+import spaces
 
 dt_str = "14062024_0910"
 
@@ -31,6 +31,7 @@ def make_spider_plot(predictions, model_names, smiles_list):
     return fig
 
 
+@spaces.GPU
 def predict_and_plot(amino_acid_sequence, smiles_input, datasets):
     model_ensemble = {}
 
@@ -128,4 +129,4 @@ iface = gr.Interface(
     theme=theme
 )
 
-iface.launch(share=True)
+iface.launch()

model/barlow_twins.py

@@ -485,6 +485,7 @@ class BarlowTwins(BaseModel):
         if self.param_dict["verbose"] is True:
             print(f"[BT]: Model saved at {folder_path}")
 
+    @spaces.GPU
     def load_model(self, path: str) -> None:
         # make weights, config and history paths
         weights_path = path + "/weights.pt"
@@ -519,7 +520,8 @@ class BarlowTwins(BaseModel):
             print(f"[BT]: Model loaded from {path}")
             print("[BT]: Loaded parameters:")
             print(self.param_dict)
-
+    
+    @spaces.GPU
     def move_to_device(self, device) -> None:
         # move each Sequential model to device
         self.enc_mol.to(device)