Fix

curegit · Apr 26, 2024 · 2927913 · 2927913
1 parent 9777a84
commit 2927913
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/descreen/cli.py b/descreen/cli.py
@@ -29,7 +29,7 @@ def main():
         # TODO
         pass
     else:
-        model = pull(args.name)
+        model = pull(args.model)
     model.to(device)
     model.eval()
     print(model)