Spaces:

GenAICoder
/

test

Sleeping

GenAICoder commited on Jan 25

Commit

ea60e87

verified ·

1 Parent(s): a73a7fe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_name = "deepseek-ai/DeepSeek-R1-Zero"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)

 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_name = "deepseek-ai/DeepSeek-R1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)