Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
import spaces
|
2 |
import gradio as gr
|
3 |
import torch
|
4 |
import numpy as np
|
@@ -13,11 +12,12 @@ import glob
|
|
13 |
import csv
|
14 |
from datetime import datetime
|
15 |
import json
|
16 |
-
from loader import load_graph_decoder
|
17 |
|
18 |
zero = torch.Tensor([0]).cuda()
|
19 |
print('outside',zero.device)
|
20 |
|
|
|
|
|
21 |
@spaces.GPU
|
22 |
def test(n):
|
23 |
print(n)
|
@@ -33,6 +33,7 @@ from rdkit import Chem
|
|
33 |
from rdkit.Chem import Draw
|
34 |
|
35 |
from evaluator import Evaluator
|
|
|
36 |
|
37 |
# Load the CSV data
|
38 |
known_labels = pd.read_csv('data/known_labels.csv')
|
|
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
import numpy as np
|
|
|
12 |
import csv
|
13 |
from datetime import datetime
|
14 |
import json
|
|
|
15 |
|
16 |
zero = torch.Tensor([0]).cuda()
|
17 |
print('outside',zero.device)
|
18 |
|
19 |
+
import spaces
|
20 |
+
|
21 |
@spaces.GPU
|
22 |
def test(n):
|
23 |
print(n)
|
|
|
33 |
from rdkit.Chem import Draw
|
34 |
|
35 |
from evaluator import Evaluator
|
36 |
+
from loader import load_graph_decoder
|
37 |
|
38 |
# Load the CSV data
|
39 |
known_labels = pd.read_csv('data/known_labels.csv')
|