Update app.py

app.py

@@ -163,10 +163,10 @@ def predict_dynamics(sequence=None, pdb_input=None, chain_id='A', use_pdb_seq=Fa
         protein = None
         if model_version == "ESM3":
             model = ESM_model(method='esm3')
-            model.load_state_dict(torch.load('Dyna-1/model/weights/dyna1.pt'), strict=False)
+            model.load_state_dict(torch.load('Dyna-1/model/weights/dyna1.pt', map_location=torch.device('cpu')), strict=False)
         else:  
             model = ESM_model(method='esm2', nheads=8, nlayers=12, layer=30).to(DEVICE)
-            model.load_state_dict(torch.load('Dyna-1/model/weights/dyna1-esm2.pt'), strict=False)
+            model.load_state_dict(torch.load('Dyna-1/model/weights/dyna1-esm2.pt', map_location=torch.device('cpu')), strict=False)
 
         model.eval()