Spaces:

sitammeur
/

ClassyText

Running

sitammeur commited on Mar 1

Commit

3120254

verified ·

1 Parent(s): 9b71853

Update src/modernbert/classifier.py

Files changed (1) hide show

src/modernbert/classifier.py CHANGED Viewed

@@ -15,6 +15,7 @@ classifier = pipeline(
     "zero-shot-classification",
     model="MoritzLaurer/ModernBERT-large-zeroshot-v2.0",
     torch_dtype=torch.bfloat16,
 )

     "zero-shot-classification",
     model="MoritzLaurer/ModernBERT-large-zeroshot-v2.0",
     torch_dtype=torch.bfloat16,
+    device="cpu"
 )