Loli-Killer commited on
Commit
988d35c
Β·
1 Parent(s): fd20a3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ model = proteinbind_new.create_proteinbind(True)
34
 
35
 
36
  def pass_through(torch_output, key: str):
37
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
38
  input_data = {
39
  key: torch_output.type(torch.float32).to(device)
40
  }
 
34
 
35
 
36
  def pass_through(torch_output, key: str):
37
+ device = torch.device("cpu")
38
  input_data = {
39
  key: torch_output.type(torch.float32).to(device)
40
  }