liuganghuggingface commited on
Commit
c5c5ff3
·
verified ·
1 Parent(s): 36123e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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')