final model

app.py

@@ -243,258 +243,6 @@ gr_interface = gr.Interface(
 
 
 if __name__ == "__main__":
-    # ensure_models_downloaded()
-    # load_model_cache()
-    gr_interface.queue(15).launch()
-
-# theme = gr.themes.Soft(
-#     primary_hue="blue",
-#     secondary_hue="slate",
-#     neutral_hue="slate",
-#     text_size="sm",
-#     spacing_size="sm",
-# ).set(
-#     background_fill_primary="*neutral_50",
-#     background_fill_secondary="*background_fill_primary",
-#     block_background_fill="*background_fill_primary",
-#     block_border_width="0px",
-#     panel_background_fill="*neutral_50",
-#     panel_border_width="0px",
-#     input_background_fill="*neutral_100",
-#     input_border_color="*neutral_200",
-#     button_primary_background_fill="*primary_300",
-#     button_primary_background_fill_hover="*primary_400",
-#     button_secondary_background_fill="*neutral_200",
-#     button_secondary_background_fill_hover="*neutral_300",
-# )
-# custom_css = """
-# #main-headertitle {
-#     text-align: center;
-#     margin-top: 15px;
-#     margin-bottom: 10px;
-#     color: var(--neutral-600);
-#     font-weight: 600;
-# }
-# #main-header {
-#     text-align: center;
-#     margin-top: 5px;
-#     margin-bottom: 10px;
-#     color: var(--neutral-600);
-#     font-weight: 600;
-# }
-# #model-settings-header, #generation-settings-header {
-#     color: var(--neutral-600);
-#     margin-top: 8px;
-#     margin-bottom: 8px;
-#     font-weight: 500;
-#     font-size: 1.1em;
-# }
-# .setting-section {
-#     padding: 10px 12px;
-#     border-radius: 6px;
-#     background-color: var(--neutral-50);
-#     margin-bottom: 10px;
-#     border: 1px solid var(--neutral-100);
-# }
-# hr {
-#     border: none;
-#     height: 1px;
-#     background-color: var(--neutral-200);
-#     margin: 8px 0;
-# }
-# #generate-btn {
-#     width: 100%;
-#     max-width: 250px;
-#     margin: 10px auto;
-#     display: block;
-#     padding: 10px 15px;
-#     font-size: 16px;
-#     border-radius: 5px;
-# }
-# #status-box {
-#     min-height: 50px;
-#     display: flex;
-#     align-items: center;
-#     justify-content: center;
-#     padding: 8px;
-#     border-radius: 5px;
-#     border: 1px solid var(--neutral-200);
-#     color: var(--neutral-700);
-# }
-# #project-badges {
-#     text-align: center;
-#     margin-top: 30px;
-#     margin-bottom: 20px;
-# }
-# #project-badges #badge-container {
-#     display: flex;
-#     gap: 10px;
-#     align-items: center;
-#     justify-content: center;
-#     flex-wrap: wrap;
-# }
-# #project-badges img {
-#     border-radius: 5px;
-#     box-shadow: 0 1px 3px rgba(0, 0, 0, 0.1);
-#     height: 20px;
-#     transition: transform 0.1s ease, box-shadow 0.1s ease;
-# }
-# #project-badges a:hover img {
-#     transform: translateY(-2px);
-#     box-shadow: 0 4px 8px rgba(0, 0, 0, 0.15);
-# }
-# #audio-output {
-#     height: 200px;
-#     border-radius: 5px;
-#     border: 1px solid var(--neutral-200);
-# }
-# .gradio-dropdown label, .gradio-checkbox label, .gradio-number label, .gradio-textbox label {
-#     font-weight: 500;
-#     color: var(--neutral-700);
-#     font-size: 0.9em;
-# }
-# .gradio-row {
-#    gap: 8px;
-# }
-# .gradio-block {
-#    margin-bottom: 8px;
-# }
-# .setting-section .gradio-block {
-#     margin-bottom: 6px;
-# }
-# ::-webkit-scrollbar {
-#   width: 8px;
-#   height: 8px;
-# }
-# ::-webkit-scrollbar-track {
-#   background: var(--neutral-100);
-#   border-radius: 4px;
-# }
-# ::-webkit-scrollbar-thumb {
-#   background: var(--neutral-300);
-#   border-radius: 4px;
-# }
-# ::-webkit-scrollbar-thumb:hover {
-#   background: var(--neutral-400);
-# }
-# * {
-#   scrollbar-width: thin;
-#   scrollbar-color: var(--neutral-300) var(--neutral-100);
-# }
-# """
-# with gr.Blocks(title="MeanAudio Generator", theme=theme, css=custom_css) as demo:
-#     gr.Markdown("# MeanAudio: Fast and Faithful Text-to-Audio Generation with Mean Flows", elem_id="main-header")
-#     badge_html = '''
-#     <div id="project-badges"> <!-- 使用 ID
-#     以便应用 CSS -->
-#     <div id="badge-container"> <!-- 添加这个容器 div 并使用 ID -->
-#         <a href="https://huggingface.co/junxiliu/MeanAudio">
-#         <img src="https://img.shields.io/badge/Model-HuggingFace-violet?logo=huggingface" alt="Hugging Face Model">
-#         </a>
-#         <a href="https://huggingface.co/spaces/chenxie95/MeanAudio">
-#         <img src="https://img.shields.io/badge/Space-HuggingFace-8A2BE2?logo=huggingface" alt="Hugging Face Space">
-#         </a>
-#         <a href="https://meanaudio.github.io/">
-#         <img src="https://img.shields.io/badge/Project-Page-brightred?style=flat" alt="Project Page">
-#         </a>
-#         <a href="https://github.com/xiquan-li/MeanAudio">
-#         <img src="https://img.shields.io/badge/Code-GitHub-black?logo=github" alt="GitHub">
-#         </a>
-#     </div>
-#     </div>
-#     '''
-#     gr.HTML(badge_html)
-#     with gr.Column(elem_classes="setting-section"):
-#         with gr.Row():
-#             available_variants = (
-#                 list(all_model_cfg.keys()) if all_model_cfg else []
-#             )
-#             default_variant = (
-#                 'meanaudio_mf'
-#             )
-#             variant = gr.Dropdown(
-#                 label="Model Variant",
-#                 choices=available_variants,
-#                 value=default_variant,
-#                 interactive=True,
-#                 scale=3,
-#             )
-#     with gr.Column(elem_classes="setting-section"):
-#         with gr.Row():
-#             prompt = gr.Textbox(
-#                 label="Prompt",
-#                 placeholder="Describe the sound you want to generate...",
-#                 scale=1,
-#             )
-#             negative_prompt = gr.Textbox(
-#                 label="Negative Prompt",
-#                 placeholder="Describe sounds you want to avoid...",
-#                 value="",
-#                 scale=1,
-#             )
-#         with gr.Row():
-#             duration = gr.Number(
-#                 label="Duration (sec)", value=10.0, minimum=0.1, scale=1
-#             )
-#             cfg_strength = gr.Number(
-#                 label="CFG (Meanflow forced to 3)", value=3, minimum=0.0, scale=1
-#             )
-#         with gr.Row():
-#             seed = gr.Number(
-#                 label="Seed (-1 for random)", value=42, precision=0, scale=1
-#             )
-#             num_steps = gr.Number(
-#                 label="Number of Steps",
-#                 value=1,
-#                 precision=0,
-#                 minimum=1,
-#                 scale=1,
-#             )
-#     generate_button = gr.Button("Generate", variant="primary", elem_id="generate-btn")
-#     generate_output_text = gr.Textbox(
-#         label="Result Status", interactive=False, elem_id="status-box"
-#     )
-#     audio_output = gr.Audio(
-#         label="Generated Audio", type="filepath", elem_id="audio-output"
-#     )
-#     generate_button.click(
-#         fn=generate_audio_gradio,
-#         inputs=[
-#             prompt,
-#             negative_prompt,
-#             duration,
-#             cfg_strength,
-#             num_steps,
-#             seed,
-#             variant,
-#         ],
-#         outputs=[generate_output_text, audio_output],
-#     )
-#     audio_examples = [
-#         ["Typing on a keyboard", "", 10.0, 3, 1, 42, "meanaudio_mf"],
-#         ["A man speaks followed by a popping noise and laughter", "", 10.0, 3, 1, 42, "meanaudio_mf"],
-#         ["Some humming followed by a toilet flushing", "", 10.0, 3, 2, 42, "meanaudio_mf"],
-#         ["Rain falling on a hard surface as thunder roars in the distance", "", 10.0, 3, 5, 42, "meanaudio_mf"],
-#         ["Food sizzling and oil popping", "", 10.0, 3, 25, 42, "meanaudio_mf"],
-#         ["Pots and dishes clanking as a man talks followed by liquid pouring into a container", "", 8.0, 3, 2, 42, "meanaudio_mf"],
-#         ["A few seconds of silence then a rasping sound against wood", "", 12.0, 3, 2, 42, "meanaudio_mf"],
-#         ["A man speaks as he gives a speech and then the crowd cheers", "", 10.0, 3, 25, 42, "fluxaudio_fm"],
-#         ["A goat bleating repeatedly", "", 10.0, 3, 50, 123, "fluxaudio_fm"],
-#         ["A speech and gunfire followed by a gun being loaded", "", 10.0, 3, 1, 42, "meanaudio_mf"],
-#         ["Tires squealing followed by an engine revving", "", 12.0, 4, 25, 456, "fluxaudio_fm"],
-#         ["Hammer slowly hitting the wooden table", "", 10.0, 3.5, 25, 42, "fluxaudio_fm"],
-#         ["Dog barking excitedly and man shouting as race car engine roars past", "", 10.0, 3, 1, 42, "meanaudio_mf"],
-#         ["A dog barking and a cat mewing and a racing car passes by", "", 12.0, 3, 5, -1, "meanaudio_mf"],
-#         ["Whistling with birds chirping", "", 10.0, 4, 50, 42, "fluxaudio_fm"],
-#     ]
-#     gr.Examples(
-#         examples=audio_examples,
-#         inputs=[prompt, negative_prompt, duration, cfg_strength, num_steps, seed, variant],
-#         #outputs=[generate_output_text, audio_output],
-#         #fn=generate_audio_gradio,
-#         examples_per_page=5,
-#         label="Example Prompts",
-#     )
-
-# if __name__ == "__main__":
-#     demo.launch()
+    ensure_models_downloaded()
+    load_model_cache()
+    gr_interface.queue(15).launch()