Upload app.py with huggingface_hub

app.py

@@ -1,6 +1,4 @@
-from graph_decoder.diffusion_model import GraphDiT
-
-import spaces
+# import spaces
 import gradio as gr
 import torch
 import torch.nn as nn
@@ -8,6 +6,9 @@ import random
 from rdkit import Chem
 from rdkit.Chem import Draw
 
+from graph_decoder.diffusion_model import GraphDiT
+
+
 ATOM_SYMBOLS = ['C', 'N', 'O', 'H']
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -23,7 +24,7 @@ path = 'model_labeled'
 def generate_random_smiles(length=10):
     return ''.join(random.choices(ATOM_SYMBOLS, k=length))
 
-@spaces.GPU
+# @spaces.GPU
 def generate_polymer(CH4, CO2, H2, N2, O2, guidance_scale):
     properties = torch.tensor([CH4, CO2, H2, N2, O2], dtype=torch.float32).unsqueeze(0)