all tensors moved to same device

app.py

@@ -143,11 +143,11 @@ def predict(model_name, pairs_file, sequence_file, progress = gr.Progress()):
         seq1 = str(seqs[prot1].seq)
         seq2 = str(seqs[prot2].seq)
 
-        fold1 = foldseek_embeddings[prot1] if do_foldseek else None
-        fold2 = foldseek_embeddings[prot2] if do_foldseek else None
+        fold1 = foldseek_embeddings[prot1].to(device) if do_foldseek else None
+        fold2 = foldseek_embeddings[prot2].to(device) if do_foldseek else None
 
-        lm1 = lm_embed(seq1)
-        lm2 = lm_embed(seq2)
+        lm1 = lm_embed(seq1).to(device)
+        lm2 = lm_embed(seq2).to(device)
 
         interaction = model.predict(lm1, lm2, embed_foldseek = do_foldseek, f0 = fold1, f1 = fold2).item()