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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def load_flan():
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
 
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