From 5bf39b4f61491255f1d95ff2615a291f0715d291 Mon Sep 17 00:00:00 2001 From: AuroraWright Date: Thu, 12 Dec 2024 21:37:14 +0100 Subject: [PATCH] Nitpick --- owocr/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/owocr/run.py b/owocr/run.py index 231033b..95c5b03 100644 --- a/owocr/run.py +++ b/owocr/run.py @@ -269,7 +269,7 @@ class TextFiltering: if torch.cuda.is_available(): device = 0 elif torch.backends.mps.is_available(): - device = "mps" + device = 'mps' else: device = -1 self.pipe = pipeline('text-classification', model=model_ckpt, tokenizer=tokenizer, device=device)