IBYDMT / app_lib /utils.py
jacopoteneggi's picture
Update
80dc74c verified
raw
history blame
384 Bytes
from huggingface_hub import hf_hub_download
supported_models_path = hf_hub_download(
repo_id="jacopoteneggi/IBYDMT",
filename="supported_models.txt",
repo_type="dataset",
)
SUPPORTED_MODELS = []
with open(supported_models_path, "r") as f:
for line in f:
line = line.strip()
model_name, _ = line.split(",")
SUPPORTED_MODELS.append(model_name)