Reality123b commited on
Commit
5879220
·
verified ·
1 Parent(s): a652850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ app = FastAPI()
12
  # Initialize text generation pipeline
13
  generator = pipeline(
14
  "text-generation",
15
- model="Qwen/Qwen3-4B-Thinking-2507",
16
  device="cpu" # Use CPU (change to device=0 for GPU)
17
  )
18
 
 
12
  # Initialize text generation pipeline
13
  generator = pipeline(
14
  "text-generation",
15
+ model="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
16
  device="cpu" # Use CPU (change to device=0 for GPU)
17
  )
18