mwzero commited on
Commit
a3b961d
·
verified ·
1 Parent(s): 3aa2ed9

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -12,7 +12,7 @@ st.title("🤖 Chatbot DeepSeek con Transformers + Streamlit")
12
  # Carica modello e tokenizer
13
  @st.cache_resource
14
  def load_model():
15
- model_name = "deepseek-ai/DeepSeek-Coder-V2-Instruct"
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
18
  return tokenizer, model
 
12
  # Carica modello e tokenizer
13
  @st.cache_resource
14
  def load_model():
15
+ model_name = "deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct"
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
18
  return tokenizer, model