Upload app.py with huggingface_hub

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from rdkit import Chem
@@ -18,6 +19,7 @@ def load_graph_decoder(path='model_labeled'):
 model = load_graph_decoder()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+@spaces.GPU
 def generate_polymer(CH4, CO2, H2, N2, O2, guidance_scale):
     properties = [CH4, CO2, H2, N2, O2]