Marcin-XStudio commited on
Commit
17c9861
·
1 Parent(s): 4b8d0c1

deepsek demo

Browse files
Files changed (2) hide show
  1. .DS_Store +0 -0
  2. app.py +1 -1
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
app.py CHANGED
@@ -15,7 +15,7 @@ from transformers import BitsAndBytesConfig, AutoModelForCausalLM
15
  load_dotenv()
16
  app = FastAPI()
17
 
18
- model_name = "numind/NuExtract-1.5"
19
 
20
  MODEL_CACHE = "/home/user/app/model_cache"
21
 
 
15
  load_dotenv()
16
  app = FastAPI()
17
 
18
+ model_name = "deepseek-ai/DeepSeek-R1"
19
 
20
  MODEL_CACHE = "/home/user/app/model_cache"
21