Upload app.py with huggingface_hub

app.py

@@ -14,17 +14,18 @@ def load_graph_decoder(path='model_labeled'):
     )
     model.init_model(path)
     model.disable_grads()
-    # model = None
     return model
 
-model = load_graph_decoder()
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# 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]
     
     try:
+        model = load_graph_decoder()
+        device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         model.to(device)
         print('enter function')
         generated_molecule, _ = model.generate(properties, device=device, guide_scale=guidance_scale)