Fix the download script on windows (#6)

2023-01-13 09:05:21 -03:00
parent d3bd6a3093
commit fcda5d7107
1 changed files with 25 additions and 23 deletions
--- a/download-model.py
+++ b/download-model.py
@@ -27,6 +27,8 @@ def get_file(args):
            f.write(data)
        t.close()

+if __name__ == '__main__':
+
    model = argv[1]
    if model[-1] == '/':
        model = model[:-1]