clementruhm commited on
Commit
37e87fa
1 Parent(s): 80ee0e5

app.py: list only CPU models from model repo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def main():
41
  with gr.Row():
42
  with gr.Column(variant="panel"):
43
  repo_files = list_repo_files(repo_id="balacoon/tts")
44
- model_files = [x for x in repo_files if x.endswith(".addon")]
45
  model_name = gr.Dropdown(
46
  label="Model",
47
  choices=model_files,
 
41
  with gr.Row():
42
  with gr.Column(variant="panel"):
43
  repo_files = list_repo_files(repo_id="balacoon/tts")
44
+ model_files = [x for x in repo_files if x.endswith("_cpu.addon")]
45
  model_name = gr.Dropdown(
46
  label="Model",
47
  choices=model_files,