Просмотр исходного кода

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

oobabooga 2 лет назад
Родитель
Сommit
bb1dac2f76
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      download-model.py

+ 1 - 1
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.