Keeby-smilyai commited on
Commit
76762cf
·
verified ·
1 Parent(s): 3206860

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -20,7 +20,7 @@ Uses **Qwen-2.5 0.5B**, CPU-friendly.
20
  # ----------------------------
21
  @st.cache_resource(show_spinner=True)
22
  def load_model():
23
- model_name = "Qwen/Qwen-2.5-0.5B"
24
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
25
  model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
26
  model.to("cpu") # CPU only
 
20
  # ----------------------------
21
  @st.cache_resource(show_spinner=True)
22
  def load_model():
23
+ model_name = "Qwen/Qwen2.5-0.5B-Instruct"
24
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
25
  model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
26
  model.to("cpu") # CPU only