Spaces:
Runtime error
Runtime error
Loli-Killer
commited on
Commit
Β·
988d35c
1
Parent(s):
fd20a3e
Update app.py
Browse files
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("
|
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 |
}
|