import gradio as gr
from stable_audio_tools import get_pretrained_model
from stable_audio_tools.interface.gradio import create_ui
import json 

import torch

def run():
    torch.manual_seed(42)

    interface = create_ui(
        model_config_path = "model_config_float_conditioning_dit_all.json", 
        ckpt_path="epoch=1201-step=560000.ckpt", 
        model_half=False
    )
    interface.queue()
    interface.launch(share=True)

if __name__ == "__main__":
    run()