Update app.py

app.py

@@ -22,7 +22,7 @@ def get_model(type='Molecule Function'):
         model.to('cuda')
     elif type == 'Biological Process':
         model.load_checkpoint("model/checkpoint_bp1.pth")
-        model.Qformer.bert.load_state_dict(torch.load('model/bp1_bert.pth').state_dict())
+        model.Qformer.bert.load_state_dict(torch.load('model/bp1_bert.pth', map_location=torch.device('cpu')).state_dict())
         model.to('cuda')
     elif type == 'Cellar Component':
         model.load_checkpoint("model/checkpoint_cc2.pth")