Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import spaces
|
2 |
import gradio as gr
|
3 |
|
|
|
4 |
import torch
|
5 |
import numpy as np
|
6 |
import pandas as pd
|
@@ -20,6 +21,7 @@ from rdkit.Chem import Draw
|
|
20 |
|
21 |
from evaluator import Evaluator
|
22 |
from loader import load_graph_decoder
|
|
|
23 |
|
24 |
# Load the CSV data
|
25 |
known_labels = pd.read_csv('data/known_labels.csv')
|
|
|
1 |
import spaces
|
2 |
import gradio as gr
|
3 |
|
4 |
+
print('device', torch.device("cuda" if torch.cuda.is_available() else "cpu"))
|
5 |
import torch
|
6 |
import numpy as np
|
7 |
import pandas as pd
|
|
|
21 |
|
22 |
from evaluator import Evaluator
|
23 |
from loader import load_graph_decoder
|
24 |
+
print('load decoder')
|
25 |
|
26 |
# Load the CSV data
|
27 |
known_labels = pd.read_csv('data/known_labels.csv')
|