liuganghuggingface commited on
Commit
ada1f2e
·
verified ·
1 Parent(s): fd8ad98

Update app.py

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