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