Convert the download option (A-Z) to upper case

2023-02-20 15:50:48 -03:00
parent d7a738fb7a
commit bb1dac2f76
1 changed files with 1 additions and 1 deletions
--- 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.