sbapan41 commited on
Commit
204f085
·
verified ·
1 Parent(s): ba23c37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,7 +26,7 @@ from qhash.model import Zonos
26
  from qhash.conditioning import make_cond_dict, supported_language_codes
27
 
28
  device = "cuda"
29
- MODEL_NAMES = ["Quantamhash/Qhash-v0.1-transformer", "Quantamhash/Qhash-v0.1-hybrid"]
30
  MODELS = {name: Zonos.from_pretrained(name, device=device) for name in MODEL_NAMES}
31
  for model in MODELS.values():
32
  model.requires_grad_(False).eval()
@@ -367,7 +367,7 @@ def build_interface():
367
  with gr.Column(scale=2):
368
  model_choice = gr.Dropdown(
369
  choices=MODEL_NAMES,
370
- value="Quantamhash/Qhash-v0.1-transformer",
371
  label="Qhash Model Type",
372
  info="Select the model variant to use.",
373
  )
 
26
  from qhash.conditioning import make_cond_dict, supported_language_codes
27
 
28
  device = "cuda"
29
+ MODEL_NAMES = ["Quantumhash/Qhash-v0.1-transformer", "Quantumhash/Qhash-v0.1-hybrid"]
30
  MODELS = {name: Zonos.from_pretrained(name, device=device) for name in MODEL_NAMES}
31
  for model in MODELS.values():
32
  model.requires_grad_(False).eval()
 
367
  with gr.Column(scale=2):
368
  model_choice = gr.Dropdown(
369
  choices=MODEL_NAMES,
370
+ value="Quantumhash/Qhash-v0.1-transformer",
371
  label="Qhash Model Type",
372
  info="Select the model variant to use.",
373
  )