Update cli.py

fakerybakery · Dec 19, 2023 · 675ac49 · 675ac49
1 parent 42b52b3
commit 675ac49
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utmos/cli.py b/utmos/cli.py
@@ -19,6 +19,7 @@ def forward(self, wav: torch.tensor) -> torch.tensor:
         output = round_down * (1. - indices.fmod(1.)).unsqueeze(0) + round_up * indices.fmod(1.).unsqueeze(0)
         return output
 
+@click.command()
 @click.argument('filename', type=click.Path(exists=True))
 def main(filepath):
     model = Score()