fix(scripts): minor fixup to gallery scripts

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-07-13 11:36:20 +02:00
parent babd5cc1a0
commit 62aa3bfdb2

View File

@ -12,7 +12,7 @@ current_dir = os.path.dirname(os.path.realpath(__file__))
def get_latest_model(): def get_latest_model():
search_term = "GGUF" search_term = "GGUF"
if sys.argv[1]: if len(sys.argv) > 2 and sys.argv[1]:
search_term = sys.argv[1] search_term = sys.argv[1]
url = "https://huggingface.co/api/models" url = "https://huggingface.co/api/models"
params = {"sort": "lastModified", "direction": -1, "limit": 30, "search": search_term} params = {"sort": "lastModified", "direction": -1, "limit": 30, "search": search_term}