Reality123b commited on
Commit
1e49580
·
verified ·
1 Parent(s): 5205e2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ app = FastAPI()
13
  generator = pipeline(
14
  "text-generation",
15
  model="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
16
- device=0 # Use CPU (change to device=0 for GPU)
17
  )
18
 
19
  # Create text streamer
 
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
 
19
  # Create text streamer