Group2255 commited on
Commit
b42d0e8
·
verified ·
1 Parent(s): 950d6d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ def load_flan():
51
 
52
  @st.cache_resource
53
  def load_deepseek():
54
- tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/deepseek-llm-7b-instruct")
55
- model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-llm-7b-instruct")
56
  return tokenizer, model
57
 
58
  @st.cache_resource
 
51
 
52
  @st.cache_resource
53
  def load_deepseek():
54
+ tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1-0528", trust_remote_code=True)
55
+ model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1-0528", trust_remote_code=True)
56
  return tokenizer, model
57
 
58
  @st.cache_resource