Hugging Face's logo

Spaces:
Malak2002
/
text-classifier-project1
Sleeping

App Files Community
sitammeur commited on
Commit
3120254
·
verified ·
1 Parent(s): 9b71853

Update src/modernbert/classifier.py

Browse files
Files changed (1) hide show
  1. src/modernbert/classifier.py +1 -0
src/modernbert/classifier.py CHANGED
@@ -15,6 +15,7 @@ classifier = pipeline(
15
  "zero-shot-classification",
16
  model="MoritzLaurer/ModernBERT-large-zeroshot-v2.0",
17
  torch_dtype=torch.bfloat16,
 
18
  )
19
 
20
 
 
15
  "zero-shot-classification",
16
  model="MoritzLaurer/ModernBERT-large-zeroshot-v2.0",
17
  torch_dtype=torch.bfloat16,
18
+ device="cpu"
19
  )
20
 
21