Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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/
|
| 55 |
-
|
| 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
|