Update src/streamlit_app.py

src/streamlit_app.py

@@ -38,12 +38,12 @@ def load_models():
 
     # Caricamento del modello Text-to-Audio (Stable Audio Open - 1.0)
     from diffusers import DiffusionPipeline
-    st.write("Caricamento del modello Stable Audio Open - 1.0 per la generazione del soundscape...")
+    st.write("Caricamento del modello Stable Audio Open Small per la generazione del soundscape...")
     # ******************** MODIFICA QUI ********************
     # Assicurati che non ci sia torch_dtype=torch.float16 e che vada sulla CPU
-    stable_audio_pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", force_download=True).to(device) 
+    stable_audio_pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-audio-open-small", force_download=True).to(device)
     # ******************************************************
-    st.write("Modello Stable Audio Open 1.0 caricato.")
+    st.write("Modello Stable Audio Open Small caricato.")
 
     return vit_gpt2_feature_extractor, vit_gpt2_model, vit_gpt2_tokenizer, stable_audio_pipeline