Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 import numpy as np
@@ -13,28 +14,6 @@ import csv
 from datetime import datetime
 import json
 
-zero = torch.Tensor([0])
-print('outside',zero.device)
-
-import spaces
-
-print('after spaces', zero.device)
-
-@spaces.GPU
-def test(n):
-    print(n)
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    print('intest device', device)
-    print('inside zero old', zero.device)
-    zero = zero.to(device)
-    print('inside zero new', zero.device)
-    # device = zero.device
-    print('1 ',torch.Tensor([1]).cuda())
-    return 0
-
-test(1)
-
-
 from rdkit import Chem
 from rdkit.Chem import Draw