diff --git a/download-model.py b/download-model.py index 5750df54..a5ee0223 100644 --- a/download-model.py +++ b/download-model.py @@ -71,7 +71,7 @@ def select_model_from_default_options(): print() print("Input> ", end='') - choice = input()[0] + choice = input()[0].strip().upper() if choice == char: print("""\nThen type the name of your desired Hugging Face model in the format organization/name.