Update src/streamlit_app.py

src/streamlit_app.py

@@ -43,12 +43,10 @@ def load_models():
     # 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",
-    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
+    cache_dir="/app/hf_cache", # Questa riga deve essere presente!
+    local_files_only=False # E questa anche!
 ).to(device)
-    # ******************************************************
-    st.write("Modello Stable Audio Open Small caricato.")
-
+    
     return vit_gpt2_feature_extractor, vit_gpt2_model, vit_gpt2_tokenizer, stable_audio_pipeline
 
 # Carica i modelli all'avvio dell'app