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 |
}
|