sergeipetrov
commited on
Commit
•
0af2224
1
Parent(s):
952ebd6
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -15,7 +15,7 @@ logger = logging.getLogger(__name__)
|
|
15 |
|
16 |
|
17 |
class EndpointHandler():
|
18 |
-
def __init__(self,
|
19 |
|
20 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
21 |
logger.info(f"Using device: {device.type}")
|
|
|
15 |
|
16 |
|
17 |
class EndpointHandler():
|
18 |
+
def __init__(self, path=""):
|
19 |
|
20 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
21 |
logger.info(f"Using device: {device.type}")
|