feat: ✨ add an headless option for model downloads

This should help with #9, i.e Google Colab mode

scripts/download_models.py

@@ -93,7 +93,7 @@ def handle_interrupt():
  console.print("Interrupted by user.", style="bold red")
 
 
-def main(models_to_download):
+def main(models_to_download, skip_input=False):
  try:
  models_to_download_selected = {}
 
@@ -129,8 +129,10 @@ def check_destination(urls, destination):
  console.print("No new models to download.")
  return
 
- models_to_download_selected = ask_user_for_downloads(
- models_to_download_selected
+ models_to_download_selected = (
+ ask_user_for_downloads(models_to_download_selected)
+ if not skip_input
+ else models_to_download_selected
  )
 
  for model_name, model_details in models_to_download_selected.items():
@@ -143,4 +145,10 @@ def check_destination(urls, destination):
 
 
 if __name__ == "__main__":
- main(models_to_download)
+ import argparse
+
+ parser = argparse.ArgumentParser()
+ parser.add_argument("-y", "--yes", action="store_true", help="skip user input")
+
+ args = parser.parse_args()
+ main(models_to_download, args.yes)