Update README.md

README.md

@@ -22,7 +22,7 @@ extra_gated_fields:
 
 # Stable Audio Open 1.0
 
-![image/webp](https://cdn-uploads.huggingface.co/production/uploads/64a22257d3149e05bc6d259f/iKlxzZY0svXP9j6j1hiWW.webp)
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/64a22257d3149e05bc6d259f/o-a_0Afsjak21fExCtkRQ.png)
 
 Please note: For commercial use, please refer to [https://stability.ai/license](https://stability.ai/license)
 
@@ -52,7 +52,7 @@ from stable_audio_tools.inference.generation import generate_diffusion_cond
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Download model
-model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
+model, model_config = get_pretrained_model("PsiPi/audio")
 sample_rate = model_config["sample_rate"]
 sample_size = model_config["sample_size"]
 
@@ -95,7 +95,7 @@ import torch
 import soundfile as sf
 from diffusers import StableAudioPipeline
 
-pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch.float16)
+pipe = StableAudioPipeline.from_pretrained("PsiPi/audio", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 
 # define the prompts