update web
Browse files
app.py
CHANGED
@@ -76,14 +76,13 @@ def load_model_if_needed(variant: str):
|
|
76 |
log.info(f"Model {variant} loaded and cached successfully")
|
77 |
return net, feature_utils
|
78 |
|
79 |
-
ensure_models_downloaded()
|
80 |
|
81 |
|
82 |
@spaces.GPU(duration=60)
|
83 |
@torch.inference_mode()
|
84 |
def generate_audio_gradio(
|
85 |
prompt,
|
86 |
-
negative_prompt,
|
87 |
duration,
|
88 |
cfg_strength,
|
89 |
num_steps,
|
@@ -179,7 +178,7 @@ variant = gr.Dropdown(label="Model Variant", choices=list(all_model_cfg.keys()),
|
|
179 |
|
180 |
gr_interface = gr.Interface(
|
181 |
fn=generate_audio_gradio,
|
182 |
-
inputs=[input_text,
|
183 |
outputs=["text", "audio"],
|
184 |
title="TangoFlux: Super Fast and Faithful Text to Audio Generation with Flow Matching and Clap-Ranked Preference Optimization",
|
185 |
description="",
|
|
|
76 |
log.info(f"Model {variant} loaded and cached successfully")
|
77 |
return net, feature_utils
|
78 |
|
79 |
+
# ensure_models_downloaded()
|
80 |
|
81 |
|
82 |
@spaces.GPU(duration=60)
|
83 |
@torch.inference_mode()
|
84 |
def generate_audio_gradio(
|
85 |
prompt,
|
|
|
86 |
duration,
|
87 |
cfg_strength,
|
88 |
num_steps,
|
|
|
178 |
|
179 |
gr_interface = gr.Interface(
|
180 |
fn=generate_audio_gradio,
|
181 |
+
inputs=[input_text, duration, cfg_strength, denoising_steps, seed, variant],
|
182 |
outputs=["text", "audio"],
|
183 |
title="TangoFlux: Super Fast and Faithful Text to Audio Generation with Flow Matching and Clap-Ranked Preference Optimization",
|
184 |
description="",
|