1 files changed, 4 insertions, 7 deletions
diff --git a/src/logreg_predct.py b/src/logreg_predct.py
index ae416b0..0f09c31 100644..100755
--- a/src/logreg_predct.py
+++ b/src/logreg_predct.py
@@ -1,11 +1,8 @@
-import sys
-
-from model import Model
+#!/bin/python3
 
+import sys
 
 if __name__ == '__main__':
-    if len(sys.argv) != 3:
-        raise 'Usage: {} dataset_path weights_path'.path(*sys.argv[1:])
-    m = Model()
-    m.predict()
+    if len(sys.argv) != 2:
+        raise 'Usage: {} dataset_path'.path(sys.argv[0])