Update app.py

app.py

@@ -1,9 +1,6 @@
-import torch
-
 import spaces
 import gradio as gr
-print('device', torch.device("cuda" if torch.cuda.is_available() else "cpu"))
-
+import torch
 import numpy as np
 import pandas as pd
 import random
@@ -22,7 +19,6 @@ from rdkit.Chem import Draw
 
 from evaluator import Evaluator
 from loader import load_graph_decoder
-print('load decoder')
 
 # Load the CSV data
 known_labels = pd.read_csv('data/known_labels.csv')
@@ -86,6 +82,15 @@ def save_interesting_log(smiles, properties, suggested_properties):
         }
         writer.writerow(log_data)
 
+print(torch.Tensor([0]).cuda())
+
+@spaces.GPU
+def test():
+    print('intest',torch.Tensor([0]).cuda())
+    return 0
+
+test()
+
 @spaces.GPU
 def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_time, model_state, num_chain_steps, fps):
     model, device = model_state