oza75
/

torchaudio-squim

Model card Files Files and versions Community

oza75 commited on 14 days ago

Commit

73a666a

·

verified ·

1 Parent(s): 67f477e

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -7,7 +7,7 @@ from io import BytesIO
 class EndpointHandler:
-    def __init__(self, **kwargs):
         """Initialize the SQUIM model handler.
         Sets up the model on GPU if available, otherwise on CPU.
         """

 class EndpointHandler:
+    def __init__(self, model_dir: str, **kwargs):
         """Initialize the SQUIM model handler.
         Sets up the model on GPU if available, otherwise on CPU.
         """