Update streamlit_app.py

src/streamlit_app.py

@@ -1,10 +1,15 @@
 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+import os
+
+# Imposta la directory di cache locale
+os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
 
 # Titolo dell'app
-st.title("🤖 Chatbot DeepSeek Transformers + Streamlit")
+st.title("🤖 Chatbot DeepSeek con Transformers + Streamlit")
 
+# Carica modello e tokenizer
 @st.cache_resource
 def load_model():
     model_name = "deepseek-ai/DeepSeek-Coder-V2-Instruct"
@@ -14,11 +19,14 @@ def load_model():
 
 tokenizer, model = load_model()
 
+# Inizializza la sessione
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 
+# Input utente
 user_input = st.text_input("Scrivi il tuo messaggio:")
 
+# Generazione risposta
 if user_input:
     st.session_state.chat_history.append(("🧑", user_input))
 
@@ -28,5 +36,6 @@ if user_input:
 
     st.session_state.chat_history.append(("🤖", response))
 
+# Mostra la conversazione
 for speaker, msg in st.session_state.chat_history:
     st.markdown(f"**{speaker}**: {msg}")