Update src/htr_pipeline/inferencer.py

src/htr_pipeline/inferencer.py

@@ -120,7 +120,7 @@ class Inferencer:
 
     @timer_func
     def transcribe_different_model(self, image, htr_tool_transcriber_model_dropdown):
-        device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
         if htr_tool_transcriber_model_dropdown == "pstroe/bullinger-general-model":
             processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")