ruslanmv commited on
Commit
10f0d10
·
verified ·
1 Parent(s): 502f9c9

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +3 -3
models.py CHANGED
@@ -1,4 +1,4 @@
1
- from transformers import pipeline
2
 
3
- def load_model():
4
- return pipeline("text-generation", model="deepseek-ai/DeepSeek-R1")
 
1
+ import os
2
 
3
+ def get_hf_api():
4
+ return "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-R1"