Update pdfextract_fun.py

pdfextract_fun.py

@@ -32,8 +32,8 @@ cfg.merge_from_file("cascade_dit_base.yml")
 cfg.MODEL.WEIGHTS = "publaynet_dit-b_cascade.pth"
 
 # Step 3: set device
-#cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-cfg.MODEL.DEVICE =  "cuda"
+cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+#cfg.MODEL.DEVICE =  "cuda"
 
 # Step 4: define model
 predictor = DefaultPredictor(cfg)