Spaces:
Paused
Paused
Update app.py
Browse files
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 = ["
|
| 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="
|
| 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 |
)
|