Spaces:
Runtime error
Runtime error
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')
|