Update src/streamlit_app.py

src/streamlit_app.py

@@ -41,7 +41,11 @@ def load_models():
     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-small", force_download=True).to(device)
+    stable_audio_pipeline = DiffusionPipeline.from_pretrained(
+    "stabilityai/stable-audio-open-small",
+    cache_dir="/app/hf_cache", # Specifica la cartella cache che hai creato nel Dockerfile
+    local_files_only=False # Lascia a False per permettere il download iniziale
+).to(device)
     # ******************************************************
     st.write("Modello Stable Audio Open Small caricato.")