Update app.py

app.py

@@ -34,8 +34,9 @@ model = proteinbind_new.create_proteinbind(True)
 
 
 def pass_through(torch_output, key: str):
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     input_data = {
-        key: torch_output,
+        key: torch_output.type(torch.float32).to(device)
     }
     output = model(input_data)
     return output[key]